注:如果使用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则表示安装成功通过