1.我的设备中这样安装,先安装CUDA_11.6.0_511.23_windows.exe,再安装cudnn-windows-x86_64-8.4.0.27_cuda11.6-archive(这个是靠移动的),在pip install pytorch和torchvision的whl文件。
2.如果需要安装paddle的,要提前看看paddle官网的安装教程,里面十分详细:
3.CUDA安装前需要安装vs2019,遇到CUDA安装错误,再来一次,哪个安装不了的就取消勾选哪一个
4.需要提前判断本机可安装哪个版本的CUDA、torch和torchvison,6中链接中含有,下面这个链接也有:https://blog.csdn.net/qq_44691524/article/details/117226459,另外,也需要判断cuda与pytorch版本对应的关系:
5.cuda历史版本安装:
cudnn历史版本安装:
6.pytorch和torchvision安装:
尽量使用whl包进行安装,因为命令行安装十分慢,whl包安装快很多,需要提前了解安装哪个版本的:其中cu表示GPU版本,cpu表示cpu版本的pytorch,cu102 # 表示 CUDA=10.2 ,cp37 # 表示 python=3.7,很好判断的,下载好之后,用pip+文件名的形式安装:
https://download.pytorch.org/whl/torch_stable.html
7.查看CUDA是否安装成功:
https://blog.csdn.net/weixin_39916758/article/details/111512561
8.查看pytorch和torchvision是否安装成功:
https://blog.csdn.net/qq_41340996/article/details/124326865
最后,本人本机的安装包如下,包含CUDA、CUDNN、pytorch、torchvison,大家可参照我的文件下自己需要的安装包:
链接:https://pan.baidu.com/s/1SVDW3-RdvAzfgmNuyAhsHQ
提取码:l45z
--来自百度网盘超级会员V3的分享