ubuntu18.04安装CUDA

ubuntu18.04安装CUDA

文章目录

  • ubuntu18.04安装CUDA
    • 1.先要安装显卡驱动
    • 2. CUDA官网下载toolkit
    • 3. 安装CUDA
    • 4. 配置环境变量
    • 5. 查看CUDA版本

1.先要安装显卡驱动

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

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

ubuntu18.04安装CUDA_第1张图片

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

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

ubuntu18.04安装CUDA_第2张图片

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

2. CUDA官网下载toolkit

进入官网,找到CUDA Toolkit/SDK

ubuntu18.04安装CUDA_第3张图片

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

ubuntu18.04安装CUDA_第4张图片

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

ubuntu18.04安装CUDA_第5张图片

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

ubuntu18.04安装CUDA_第6张图片

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

ubuntu18.04安装CUDA_第7张图片

下载命令

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,需要下载很久。

3. 安装CUDA

sudo sh cuda_11.2.0_460.27.04_linux.run

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

ubuntu18.04安装CUDA_第8张图片

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

ubuntu18.04安装CUDA_第9张图片

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

ubuntu18.04安装CUDA_第10张图片

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

4. 配置环境变量

打开 /home/username/.bashrc 文件,最后添加两行:

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

5. 查看CUDA版本

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

nvcc -V

ubuntu18.04安装CUDA_第11张图片
安装成功!

你可能感兴趣的:(Linux,深度学习,pytorch,caffe)