conda虚拟环境内安装CUDA9.0

1.下载CUDA9.0
下载地址
如果ubuntu版本过高，则需在conda环境内安装指定版本的gcc，可以参考这里。
选择runfile下载，下载完成之后执行如下语句，运行.run文件

sh cuda_9.0.176_384.81_linux.run

单击回车，一路往下运行，直到提示“是否为NVIDIA安装驱动nvidia-384？”，选择否，因为已经安装好驱动程序了，其他的全都是默认，不过要记住安装位置，默认是安装在/usr/local/cuda文件夹下。

2.在conda环境内设置环境变量

• 使用conda env config vars set my_var=value设置当前虚拟环境中的自定义环境变量，这里使用以下命令更改所有虚拟环境内的cuda环境变量

 conda env config vars set LD_LIBRARY_PATH=/mypath/cuda90/lib64
 conda env config vars set PATH=/mypath/cuda90/bin:$PATH
 conda env config vars set CUDA_HOME=/mypath/cuda90

• 通过-n参数可以指定具体的虚拟环境来设置自定义环境变量，比如下面：

 conda env config vars set LD_LIBRARY_PATH=/mypath/cuda90/lib64 -n my_env
 conda env config vars set PATH=/mypath/cuda90/bin:$PATH -n my_env
 conda env config vars set CUDA_HOME=/mypath/cuda90 -n my_env

这样，只有进入my_env的环境时会默认使用cuda90，其余环境不受影响。

• 设置完环境变量后必须重新激活环境`

 conda activate my_env

可以使用

conda env config vars list

来查看虚拟环境中的环境变量

3.安装cudnn

cp cudnn-9.0-linux-x64-v7.4.1.5.solitairetheme8 cudnn-9.0-linux-x64-v7.4.1.5.tgz
tar -xvf cudnn-9.0-linux-x64-v7.4.1.5.tgz

随后移到cuda90的目录下

cp cuda/include/cudnn.h /mypath/cuda90/include
cp cuda/lib64/libcudnn* /mypath/cuda90/lib64

查看cudnn版本

cat /mypath/cuda90/include/cudnn.h | grep CUDNN_MAJOR -A 2