1. 程式人生 > >非root使用者安裝CUDA和CuDNN

非root使用者安裝CUDA和CuDNN

這裡寫自定義目錄標題

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