更换cudnn内容

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
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值