1.下载CUDA
在以下网址下载9.0的deb文件(注意选取合适的平台):
https://developer.nvidia.com/cuda-downloads
2.终端安装deb文件
在终端执行如下命令:
sudo dpkg -i cuda-repo-ubuntu1604-9-0-local_9.0.176-1_amd64.deb sudo apt-key add /var/cuda-repo-<version>/7fa2af80.pub #执行完上一句指令后,末尾会有此指令的提示 sudo apt-get update sudo apt-get install cuda
3.修改环境变量
sudo gedit ~/.bashrc将如下内容添加到.bashrc文件末尾:
export CUDA_HOME=/usr/local/cuda-9.0
export LD_LIBRARY_PATH=${CUDA_HOME}/lib64
export PATH=${CUDA_HOME}/bin:${PATH}
修改完成后,在终端输入指令,使环境变量立即生效:
sudo ldconfig
4.验证
在终端输入命令,实时查看GPU的使用情况:
watch -n 1 nvidia-smi原文链接:https://www.f2er.com/ubuntu/350894.html