1. 程式人生 > >CUDA和cudnn的環境變數設定問題

CUDA和cudnn的環境變數設定問題

在 ~/.bashrc 裡再設定一下cuda的環境變數:

export PATH=/usr/local/cuda/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
export CUDA_HOME=/usr/local/cuda

同時 source ~/.bashrc 讓其生效。

參考: