CUDA 9.0在Ubuntu上的安装

前端之家收集整理的这篇文章主要介绍了CUDA 9.0在Ubuntu上的安装前端之家小编觉得挺不错的,现在分享给大家,也给大家做个参考。

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.修改环境变量

在终端打开并修改.bashrc文件

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

猜你在找的Ubuntu相关文章