非root使用者安裝CUDA和CuDNN
阿新 • • 發佈:2018-12-21
這裡寫自定義目錄標題
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