一、Anconda的安装
1.官方网站:Anaconda | The World's Most Popular Data Science Platform
然后无脑安装就可以了
二、安装CUDA
1.查看本机cuda适配版本
打开NVIDIA控制面板(可以在任务栏小图标右键或者初始菜单中找到),选择帮助,点击系统支持
这里可以看到我的最高版本是11.2,所以应该选择不高于11.2的cuda版本
如果安装失败,就选择更低版本的就行
2.进入NVIDIA官网CUDA Toolkit Archive | NVIDIA Developer
选择合适的版本下载,程序类型选择local,之后无脑安装就行了。
3.验证是否安装成功
菜单栏打开Anaconda Powershell prompt,输入nvcc -V,如果成功会返回版本号
三、cuDNN下载
1.官方网站:https://developer.nvidia.com/rdp/cudnn-download
(下载需要登陆NVIDIA账号)
2.打开CUDA文件目录,把下载完成后的压缩包解压,并把里面的文件复制到cuda相应文件夹的目录下。
四、PyTorch的安装(需要科学上网)
1.官方网站:Start Locally | PyTorch
2.选择如下,并复制这串代码
4.打开Anaconda prompt,粘贴这串代码并回车,然后等待安装即可
五、验证安装是否成功
1.在anaconda prompt中输入python
然后输入import torch
print(torch.cuda.is_available())
返回True就说明cuda安装完毕
2.输入from torch.backends import cudnn
print(cudnn.is_available())
返回True就说明安装完毕



