第一、二步 安装cuda、cuDNN
先安装cuda,cuDNN,这两步在下面这个链接中讲的很详细,感谢博主分享经验(23条消息) 安装CUDA,cuDNN,Pytorch的详细教程,一气呵成!_Nefu_lyh的博客-CSDN博客
这两步基本按他的流程走完就可以了,我想补充的是:
① cuda一定安装路径一定要在默认位置!不要强迫症地去改他!
②安装完cuda把下面这几个环境变量给加上(其中CUDA_PATH和CUDA_PATH_V10_2是安装cuda的时候自动加上的。而且Path中也会自动加上两个路径
一开始加上后我在cmd中nvcc -V还是说没有找到该命令,重启后就好了。
第三步 安装pytorch
主要说一下第三步,安装pytorch。链接中的方法我试了仍然出现read time out的报错,最终用的是pip换清华源直接一步安装,比链接中分别安装三个文件要方便得多。
这里最重要的是版本匹配问题,
- cuda cudnn NvidiaDriver版本必须匹配
- torch torchaudio torchvisio