本文共 763 字，大约阅读时间需要 2 分钟。

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  

转载地址：http://zoqhi.baihongyu.com/