0.确认你的电脑上有英伟达显卡
win+R
输入:control /name Microsoft.DeviceManager
显示适配器
,能看到显卡即可。1.查看主机显卡驱动版本. 主机GPU驱动版本决定你的主机最高能支持到什么版本的cuda程序
2.根据主机显卡驱动版本来确定主机应该安装哪个版本的CUDA Tookit
Table 3 CUDA Toolkit and Corresponding Driver Versions
546.12
,所以我的电脑最高支持的版本为: CUDA 12.2 GA
3.下载 CUDA Tookit
Archive of Previous CUDA Releases
,选一个版本下载4.设置环境变量
CUDA_PATH : C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.8
%CUDA_PATH%
以及 %CUDA_PATH%\bin
nvcc --version
检测是否成功有两种方法下载安装。一种是通过上文提到的手动下载并安装。还有一种是通过conda命令行下载并安装。
在这里我使用conda命令行进行演示:
1.下载安装 Anaconda
2.安装指定版本cuda(11.8)
conda install cuda -c nvidia/label/cuda-11.8.0
3.检查cuda是否安装完毕
import torch
print(torch.cuda.is_available())
print(torch.__version__) #注意是双下划线
1.pytorch和cudatoolkit版本对应关系
2.cudatoolkit版本和系统cuda对应关系:
3.系统cuda和nvidia对应关系:
4.cuda和cuDNN的关系和对应关系:
nvidia-smi
,找到CUDA Version
CUDA版本和显卡驱动版本对照表
显卡驱动下载
pytorch官方
官网-安装cuda教程
参考安装文章1
参考安装文章2
参考安装文章3
参考安装文章4
参考安装文章5 -环境变量
------ 如果文章对你有用,感谢右上角 >>>点赞 | 收藏 <<<