注:如果使用anaconda,貌似不需要手動安裝Cuda和cudnn,安裝tensorflow時會自動安裝
1.官方教程https://docs.nvidia.com/cuda/cuda-installation-guide-linux/index.html#ubuntu-installation
2.在http://developer.nvidia.com/cuda-downloads上下載安裝包
3.到安裝文件目錄下運行.run文件,輸入accept
sudo sh cuda_10.1.168_418.67_linux.run
4.我之前已經安裝了Nvidia的顯卡驅動,這里不安裝driver
因為我們只是調用CUDA,不去寫CUDA程序,所以Samples也可以不安裝
5.安裝完成
6.添加環境變量
vi ~/.bashrc
在文件末尾添加
export PATH="/usr/local/cuda-10.1/bin:$PATH"
export LD_LIBRARY_PATH="/usr/lcoal/cuda-10.1/lib64:$LD_LIBRARY_PATH"
最后使其生效
source ~/.bashrc
7.終端輸入
cd /usr/local/cuda-10.1/samples/1_Utilities/deviceQuery
sudo make
./deviceQuery
結果如圖
出現Result = PASS則表示安裝成功通過