一.先看自己的电脑NVIDIA 支持CUDA版本是多少?
1.打开NVIDIA控制面板
2.点击帮助---系统信息--组件
我的支持CUDA11.6
二.再看支持Pytorch的CUDA版本
三.打开CUDA官网
下载CUDA 11.6
下载好后,安装
选择 自定义 然后安装位置 (先去F盘 建个CUDA-manger文件夹 然后在里面建个CUDA11.6文件夹 再在里面建立CUDA1 CUDA2这两个文件夹
前两个位置选到CUDA1
第三个选择CUDA2
然后下一步
最后验证 在anaconda promt里输入nvcc -V 可以看到CUDA的版本
四.然后去下载Cudnn,去官网
没有账号的需要注册一个,然后登陆,下载CUDA11.6对应的Cudnn,下载好之后,解压在F盘
五,安装Pytorch
找到对应的版本