cuda 8.0 /9.0软连接切换
原创 aNLP大虾 最后发布于2019-09-30 10:09:35 阅读数 138 收藏
展开
目录组成:
在这里插入图片描述
1. cuda的切换
1、删除软连接:
@ASR:/usr/local$ sudo rm -rf cuda
rm -rf /usr/local/cuda #删除之前创建的软链接 绝对路径版本
2、增加软连接
@ASR:/usr/local$ sudo ln -s /usr/local/cuda-9.0 cuda
sudo ln -s /usr/local/cuda-8.0 /usr/local/cuda
2. cudnn的切换
cudnn的切换与安装一致,只要把cudnn的库复制到cuda的目录下即可。
sudo cp cuda/include/cudnn.h /usr/local/cuda/include/
sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64/
sudo chmod a+r /usr/local/cuda/include/cudnn.h
sudo chmod a+r /usr/local/cuda/lib64/libcudnn*
查看cudnn的版本: 去cudnn的头文件打印即可
cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2
3、查看切换flag
ls -l cuda
如:cuda -> /usr/local/cuda-9.0表示成功
最后按照pytorch
conda install pytorch=1.1.0 torchvision cudatoolkit=9.0 -c pytorch