【亲测】ubuntu20.4 安装CUDA11.8

首先回答以下为什么选择安装cuda11.8,原因是目前pytorch架构最高支持11.8版本,所以为了更好的使用pytorch,我安装cuda11.8.

一、在NVIDIA官网下载CUDA11.8

1、网址:CUDA Toolkit Archive | NVIDIA Developer

2、选择CUDA11.8

【亲测】ubuntu20.4 安装CUDA11.8_第1张图片

3、选择安装系统信息

【亲测】ubuntu20.4 安装CUDA11.8_第2张图片

4、这里Installer Type我们选择deb(local)方式安装(下载速度快.......),此时会出现安装步骤:

【亲测】ubuntu20.4 安装CUDA11.8_第3张图片

5、以此按上面的提示进行安装即可(安装过程中有点耗时、有点卡,请耐心等待)

二、配置环境

1、sudo gedit ~/.bashrc

2、在打开的 .bashrc最后添加

export PATH=$PATH:/usr/local/cuda-11.8/bin
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda-11.8/lib64
export LIBRARY_PATH=$LIBRARY_PATH:/usr/local/cuda-11.8/lib64

3、配置环境生效

source ~/.bashrc

三、查看CUDA安装信息

1、查看版本号

nvcc --version

【亲测】ubuntu20.4 安装CUDA11.8_第4张图片

此刻CUDA11.8安装成功!

四、卸载CUDA11.8

1、sudo apt-get --purge remove cuda*

2、sudo apt-get autoremove

3、sudo reboot

你可能感兴趣的:(C++,C++,ONNX推理部署,CUDA11.8,ubuntu,linux,opencv,人工智能,神经网络,YOLO)