一些代碼運行的時候需要GPU來加速,cuda安裝之後,或者使用root賬戶安裝好的公用的cuda,當前用戶需要在自己主目錄下**.bashrc** 正確配置要使用的cuda路徑
- cuda配置如下:
#<<< config root's cuda start <<<
export PATH=/usr/local/cuda-8.0/bin:$PATH
export CUDA_HOME=/usr/local/cuda-8.0/bin:$CUDA_HOME
export LD_LIBRARY_PATH=/usr/local/cuda-8.0/lib64:/usr/local/cuda-8.0/extras/CUPTI/lib64:$LD_LIBRARY_PATH
#<<< config root's cuda end <<<
- .bashrc 環境變量配置文件在當前用戶主目錄下:
vim .bashrc
把上面cuda的配置copy到最下面,指定你要使用的cuda的目錄,即可
然後使用source命令使配置生效
source .bashrc
root 賬戶 source命令方式:
source ~/.bashrc