cuda如何做软连接切换

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
 

扫描二维码关注公众号，回复： 9748521 查看本文章