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