这里写自定义目录标题
Linux驱动程序安装
最近一直在机器上跑pytorch,无奈公司的机器没有root权限,一直有版本不匹配的情况,所以自己又重新安装了一下CUDA和CuDNN的版本,这次以CUDA8.0和CuDNN7.0为例
CUDA
下载安装包cuda_8.0.61_375.26_linux.run
然后运行
sh cuda_8.0.61_375.26_linux.run
然后选择接受EULA,accept,但是后面拒绝安装驱动no,然后是安装目录,这时选择自己的安装目录/home/xxx/cuda
CuDNN
下载CuDNN文件cudnn-8.0-linux-x64-v7.tgz
tar -zvxf cudnn-8.0-linux-x64-v7.tgz
然后把里面的文件copy到cuda目录下
cp include/cudnn.h /data1/xxxxx/opt/cuda-8.0/include/
cp lib64/libcudnn* /data1/xxxxx/opt/cuda-8.0/lib64/
chmod a+r include/cudnn.h lib64/libcudnn*
然后更改环境变量
export CUDA_HOME="/data1/supinyu/opt/cuda-8.0"
export PATH="$PATH:$CUDA_HOME/bin"
export LD_LIBRARY_PATH="$LD_LIBRARY_PATH:$CUDA_HOME/lib64"
然后保存环境变量
source .bashrc