安裝cuda配置環境變量

打開bash_profilepython

sudo gedit ~/.bash_profile

在打開的文本末尾加入:vim

export LD_LIBRARY_PATH="$LD_LIBRARY_PATH:/usr/local/cuda/lib64:/usr/local/cuda/extras/CUPTI/lib64"
export CUDA_HOME=/usr/local/cuda

 保存退出後經過以下命令使環境變量生效:bash

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/82753403spa

相關文章
相關標籤/搜索