2、点击安装程序进行安装
选择默认路径即可(只是个临时提取安装程序的文件夹)
选择自定义安装
将Visual Studio Integration选项取消(没什么用而且会影响下载)
确定安装路径(可以修改,最好记住)
等待安装即可
安装完成后查看一下是否有环境变量,没有自己手动添加
CUDA_PATH
CUDA_PATH_V12_3
测试环境是否安装成功
打开cmd,输入
nvcc -V
查看cuda版本
输入
set cuda
查看环境变量
如上两图即为下载成功!
二、安装Pytorch
1、了解对应的pytorch版本
要成功运行cuda架构,所需的pytorch版本必须与python和cuda版本对应,以下为cuda与pytorch对应关系
cuda与pytorch版本对应表
CUDA版本 | 可用PyTorch版本 |
---|