安装cuda到某个路径下,然后修改个人环境变量来指定这个cuda路径。这样就可以换成新版本的CUDA并且不影响服务器主环境。
第一步,下载cuda安装包:
完成下载后,运行run文件
./cuda*.run
注意,是否安装Accelerated Graphics Driver,选择no。如下:
第二步,下载cudnn
解压文件
tar -xzvf cudnn*.tgz
把下列文件拷到cuda安装目录下
cp cuda/include/cudnn.h cudaa/include/
cp cuda /lib64/libcudnn* cudaa/lib64
chmod a+r cuda/include/cudnn/h cudaa/lib64/libcudnn*
第三步:修改bashrc
在bashrc中加入:
export PATH="/home/liweilia/cudaa/bin:$PATH"
export LD_LIBRARY_PATH="$LD_LIBRARY_PATH:/home/cudaa/lib64/"
然后source .bashrc
p.s. 还没有亲测,以后有时间试下
网友评论