当前位置:   article > 正文

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

ubuntu cuda11.8

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

一、在NVIDIA官网下载CUDA11.8

1、网址:CUDA Toolkit Archive | NVIDIA Developer

2、选择CUDA11.8

3、选择安装系统信息

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

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

此刻CUDA11.8安装成功!

四、卸载CUDA11.8

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

2、sudo apt-get autoremove

3、sudo reboot

声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/我家小花儿/article/detail/972512
推荐阅读
相关标签
  

闽ICP备14008679号