打開bash_profile:
sudo gedit ~/.bash_profile
在打開的文本末尾加入:
export LD_LIBRARY_PATH="$LD_LIBRARY_PATH:/usr/local/cuda/lib64:/usr/local/cuda/extras/CUPTI/lib64" export CUDA_HOME=/usr/local/cuda
保存退出后通過如下命令使環境變量生效:
source ~/.bash_profile
還看到一個方法:
sudo vim ~/.bashrc
按i按鍵:
來編輯.bashrc配置文件,到文件最后,寫入下面幾句話(直接復制就行) export PATH=/usr/local/cuda-9.0/bin:$PATH export LD_LIBRARY_PATH=/usr/local/cuda-9.0/lib64$LD_LIBRARY_PATH
Ubuntu18.04下安裝深度學習框架Pytorch(GPU加速)
https://blog.csdn.net/wuzhiwuweisun/article/details/82753403