ubuntu18.04安装CUDA

moneymyone

已于 2022-01-27 09:01:42 修改

阅读量8.3k

点赞数 7

分类专栏： Linux 文章标签：深度学习 pytorch caffe

于 2022-01-27 08:59:27 首次发布

本文链接：https://blog.youkuaiyun.com/moneymyone/article/details/122711846

版权

31 篇文章

订阅专栏

可通过ubuntu Software & Updates中的Additional Drivers安装

比如可选择460或470驱动，笔者选择了460。此步骤不装也行，CUDA安装包中包含安装驱动。

[外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-vOCf2k4M-1643244750244)(../../../Image-Typora/image-20220126161435419.png)]

点击Apply Changes可进行安装,安装完可输入下面命令查看是否安装成功

nvidia-smi # 查看是否安装成功

[外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-2yq4B89q-1643244750245)(../../../Image-Typora/image-20220126161657634.png)]

图中460已经安装成功，CUDA Version: 11.2表示可安装的CUDA最高版本为11.2，所以进入CUDA官网时，一定要选择该版本，否则会安装不成功。

进入官网，找到CUDA Toolkit/SDK

[外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-UKyqcmC1-1643244750246)(../../../Image-Typora/image-20220126162104108.png)]

进入Toolkit下载页面，点击Download Now>

[外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-LQY7wyuL-1643244750247)(../../../Image-Typora/image-20220126162440971.png)]

这部不要直接选择系统类型，选择下面的Archive of Previous CUDA Releases

[外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-My1cN2EK-1643244750247)(../../../Image-Typora/image-20220126162417335.png)]

选择历史版本下载，笔者下载 CUDA Toolkit 11.2.0

[外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-HpYgsuaE-1643244750248)(../../../Image-Typora/image-20220126162745096.png)]

选择系统版本等信息，获取下载链接

[外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-foWNFmIU-1643244750248)(../../../Image-Typora/image-20220126162855097.png)]

下载命令

wget https://developer.download.nvidia.com/compute/cuda/11.2.0/local_installers/cuda_11.2.0_460.27.04_linux.run

cuda_11.2.0_460.27.04_linux.run 共3G，需要下载很久。

sudo sh cuda_11.2.0_460.27.04_linux.run

程序起来之后，会出现下面的提示，程序检测到电脑里面已经安装了驱动了，但是cuda还没安装，仍然继续，点击Continue

[外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-ihMcLzBp-1643244750249)(../../../Image-Typora/image-20220126165204890.png)]

会显示是否同意License的界面，输入accept

[外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-8IQzVjy7-1643244750249)(../../../Image-Typora/image-20220126165236466.png)]

进入到CUDA 安装项选择界面，因为已经安装了Nvidia的驱动460，这一步跳过，不要勾选Driver，注意：X表示选择，空的中括号表示不选择。

[外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-QghSIlyF-1643244750250)(../../../Image-Typora/image-20220126165250703.png)]

安装过程很快，2-3分钟。

打开 /home/username/.bashrc 文件，最后添加两行：

export PATH=$PATH:/usr/local/cuda/bin
export CUDA_HOME=$CUDA_HOME:/usr/local/cuda

安装好CUDA并配置好环境变量后，重启终端或另打开一个终端输入：

nvcc -V

[外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-P1q9DDBU-1643244750251)(../../../Image-Typora/image-20220127084959185.png)]
安装成功！