pytorch和cuda版本不同需要的cudnn版本就不一样,在每次运行前注意以下所用的cudnn版本
pytorch-0.4.1-py35_cuda9.2.148_cudnn7.1.4_1
pytorch-0.4.1-py36_cuda9.2.148_cudnn7.1.4_1
pytorch-1.2.0-py3.6_cuda9.2.148_cudnn7.6.2_0
pytorch-1.3.0-py3.6_cuda9.2.148_cudnn7.6.3_0
pytorch-1.3.0-py3.6_cuda10.1.243_cudnn7.6.3_0
pytorch-1.4.0-py3.6_cuda9.2.148_cudnn7.6.3_0
pytorch-1.4.0-py3.6_cuda10.1.243_cudnn7.6.3_0
#9.2-cudnn7.1.4
sudo cp /home/heixiazi/add/cudnn/cuda9.2-7.1.4/include/cudnn.h /usr/local/cuda-9.2/include
sudo cp /home/heixiazi/add/cudnn/cuda9.2-7.1.4/lib64/libcudnn* /usr/local/cuda-9.2/lib64
sudo chmod a+r /usr/local/cuda-9.2/include/cudnn.h /usr/local/cuda-9.2/lib64/libcudnn*
source ~/.bashrc
#9.2-cudnn7.6.3
sudo cp /home/heixiazi/add/cudnn/cuda9.2-7.6.3/include/cudnn.h /usr/local/cuda-9.2/include
sudo cp /home/heixiazi/add/cudnn/cuda9.2-7.6.3/lib64/libcudnn* /usr/local/cuda-9.2/lib64
sudo chmod a+r /usr/local/cuda-9.2/include/cudnn.h /usr/local/cuda-9.2/lib64/libcudnn*
source ~/.bashrc
#9.2-cudnn7.6.2
sudo cp /home/heixiazi/add/cudnn/cuda9.2-7.6.2/include/cudnn.h /usr/local/cuda-9.2/include
sudo cp /home/heixiazi/add/cudnn/cuda9.2-7.6.2/lib64/libcudnn* /usr/local/cuda-9.2/lib64
sudo chmod a+r /usr/local/cuda-9.2/include/cudnn.h /usr/local/cuda-9.2/lib64/libcudnn*
source ~/.bashrc
#10.1-cudnn7.6.3
sudo cp /home/heixiazi/add/cudnn/cuda10.1-7.6.3/include/cudnn.h /usr/local/cuda-10.1/include
sudo cp /home/heixiazi/add/cudnn/cuda10.1-7.6.3/lib64/libcudnn* /usr/local/cuda-10.1/lib64
sudo chmod a+r /usr/local/cuda-10.1/include/cudnn.h /usr/local/cuda-10.1/lib64/libcudnn*
source ~/.bashrc
验证
#9.2
cd /usr/local/cuda-9.2/samples/1_Utilities/deviceQuery
sudo make
./deviceQuery
1211

被折叠的 条评论
为什么被折叠?



