Ubuntu下安装、配置及重装CUDA教程

这篇具有很好参考价值的文章主要介绍了Ubuntu下安装、配置及重装CUDA教程。希望对大家有所帮助。如果存在错误或未考虑完全的地方,请大家不吝赐教,您也可以点击"举报违法"按钮提交疑问。

CUDA

前往Nvidia CUDA Tools官网选择对应的架构和版本下载CUDA

以如下架构和版本为例:
Ubuntu下安装、配置及重装CUDA教程,ubuntu,linux,运维

查看显卡驱动

nvidia-smi

如果显卡驱动已经装了(例如此处提示的Driver Version: 535.54.03),那么在CUDA安装过程中选择需要安装的组件时,不用再勾选安装driver。
Ubuntu下安装、配置及重装CUDA教程,ubuntu,linux,运维

下载并安装CUDA

根据刚才选择的的操作系统、架构等对应的下载安装命令来下载并安装CUDA

wget https://developer.download.nvidia.com/compute/cuda/11.8.0/local_installers/cuda_11.8.0_520.61.05_linux.run
sudo sh cuda_11.8.0_520.61.05_linux.run

添加环境变量

sudo nano ~/.bashrc
# Add
export PATH=/usr/local/cuda-11.8/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda-11.8/lib64:$LD_LIBRARY_PATH
export CUDA_HOME=/usr/local/cuda

卸载CUDA

常见的情况是,在使用Pytorch时对CUDA版本有要求,即使在Conda环境下也偶尔会有问题,因此降级到Pytorch2.0.1对应的CUDA11.8,假设此时cuda版本为12.1

cd /usr/local/cuda-12.1/bin/
sudo ./cuda-uninstaller
sudo rm -rf /usr/local/cuda-12.1

Conda

下载Anaconda

前往Anaconda官方下载页下载安装脚本。

对于Nvidia Orin AGX:

wget https://repo.anaconda.com/archive/Anaconda3-2023.09-0-Linux-aarch64.sh
bash Anaconda3-2023.09-0-Linux-aarch64.sh

安装Anaconda

赋权

sudo chmod +x Anaconda3-2023.07-2-Linux-x86_64.sh

执行脚本

./Anaconda3-2023.07-2-Linux-x86_64.sh

注意: 如果不希望启动命令行终端自动打开conda base,记得不要在安装时设置base环境自动激活
如果不慎设置了自动激活base
可以输入以下指令

conda config --set auto_activate_base false

检查安装是否完成

conda --version

创建环境

conda create -n <env_name> python=3.7 -y

删除环境

conda remove -n <env_name> --all

激活环境

conda activate <env_name>

退出环境

conda deactivate <env_name>

安装软件

在对应的虚拟环境内使用安装

conda install <app_name>

指定channel安装

conda install -c conda-forge package-name

查看conda信息

conda info

查看conda的环境变量设置

conda env config vars list
conda env config vars set

卸载软件

在对应的虚拟环境内使用卸载

conda uninstall <app_name>

Pytorch

前往Pytorch官方网站选择合适的Pytorch版本并下载

conda install pytorch torchvision torchaudio pytorch-cuda=11.8 -c pytorch -c nvidia

检查安装是否成功文章来源地址https://www.toymoban.com/news/detail-581131.html

conda list | grep pytorch
(base) hermanye@hermanye:~$ python
Python 3.11.4 (main, Jul  5 2023, 14:15:25) [GCC 11.2.0] on linux
Type "help", "copyright", "credits" or "license" for more information.
>>> import torch
>>> x = torch.rand(5, 3)
>>> print(x)
tensor([[0.5421, 0.5950, 0.3337],
        [0.8443, 0.2287, 0.5316],
        [0.0301, 0.0151, 0.3522],
        [0.3456, 0.5901, 0.5970],
        [0.6271, 0.8065, 0.7645]])
>>> 

>>> torch.cuda.is_available()
True
>>> torch.cuda.current_device()
0
>>> torch.cuda.device(0)
<torch.cuda.device object at 0x7f2bb0556700>
>>> torch.cuda.get_device_name(0)
'NVIDIA GeForce RTX 4070 Ti'

>>> print(torch.__version__)
1.8.2+cu111
>>> print(torch.version.cuda)
11.1

到了这里,关于Ubuntu下安装、配置及重装CUDA教程的文章就介绍完了。如果您还想了解更多内容,请在右上角搜索TOY模板网以前的文章或继续浏览下面的相关文章,希望大家以后多多支持TOY模板网!

本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如若转载,请注明出处: 如若内容造成侵权/违法违规/事实不符,请点击违法举报进行投诉反馈,一经查实,立即删除!

领支付宝红包 赞助服务器费用

相关文章

  • 问题7:虚拟机+ubuntu+安装cuda(傻瓜式操作)+cuda path配置+查看cuda的版本

    目录 1.cuda的安装 2.cuda path的配置 3.检查cuda的版本号(为安装pytorch做准备) ... 建议看看下面的几条ps,可以避免踩坑! ps:本文所用ubuntu系统版本为v-22.04(如果打算安装可以参考博主的另一篇文章),ubuntu-v-22.04最高支持cuda-v-11.7.0 ps:此文为2023.2.4所写,此时pytorch支持的c

    2024年02月02日
    浏览(39)
  • Ubuntu20.04安装CUDA&&cudnn(初学者详细图文教程)

    注:本机系统为 Ubuntu20.04 1. 安装显卡驱动 打开‘软件和更新,点击附加驱动安装显卡驱动。 2. gcc安装 若系统为Ubuntu22.04,则需要安装。 在终端输入一下指令,查看有没有gcc。 3. gcc安装 本机ubuntu20.04 gcc自带版本为9.4.0 安装gcc-7,command: 安装完gcc-7,系统中就存在两个版本的

    2023年04月12日
    浏览(52)
  • 【Ubuntu20.04 CUDA11.1+Torch1.10+Anaconda 保姆级安装教程】

    安装CUDA时需要和Torch版本对应起来,最好先去torch官网上确定要安装的torch版本对应的CUDA版本。 在安装CUDA之前需要先确定是否已经安装驱动,打开终端输入 nvidia-smi ,若有输出,则表明驱动安装过,否则需要先安装驱动(驱动安装教程) 1、CUDA11.1下载 先去CUDA官网上下载要安

    2024年02月02日
    浏览(67)
  • 【PaddlePaddle】保姆级教程:Ubuntu22.04+CUDA12.2编译PaddlePaddle源码并安装

    恰逢研一开学,在上课之余想利用这可以自己支配的半年时间再扩展一项技能。 突然想到自己平常在某乎看到众多帖子均在写pytorch大大小小的坑,于是思来想去,决定支持一下国产深度学习框架:PaddlePaddle 开个新坑,记录自己的学习历程 需要先启用“适用于 Linux 的 Window

    2024年02月03日
    浏览(56)
  • ubuntu20.04显卡驱动cuda cudnn conda TensorRT安装及配置

    如果要使用 TensorRT ,请注意CUDA 支持的最高算力要大于等于GPU本身的算力, 可以先看3小节 conda和pip换源直接看2.3小节 本人已在ubuntu20.04下安装成功。其他版本步骤应该差不多 如果帖子有帮助,感谢一键三连, ^_^ 部署有问题的小伙伴欢迎留言和加 Q 裙- 472648720 BEV各算法环境部

    2024年01月22日
    浏览(57)
  • 超详细教程——Ubuntu20.04 安装英伟达NVIDIA显卡驱动、CUDA、Cmake以及不同版本的CUDA切换

    在windows和linux双系统下,为ubuntu安装nvidia、cuda、cmake等程序是一个可能会难到各个计算机小白的问题,本文将一步步地带您学会安装以上内容。 一、安装英伟达NVIDIA显卡驱动 对于新安装的linux操作系统,我们需要为它安装nvidia显卡驱动(ubuntu有自带的显卡驱动但更推荐安装

    2024年02月02日
    浏览(57)
  • ubuntu18.04复现yolo v8环境配置之CUDA与pytorch版本问题以及多CUDA版本安装及切换

    最近在复现yolo v8的程序,特记录一下过程 环境:ubuntu18.04+ros melodic 小知识:GPU并行计算能力高于CPU—B站UP主说的 Ubuntu可以安装多个版本的CUDA。如果某个程序的Pyorch需要不同版本的CUDA,不必删除之前的CUDA,可以实现多版本的CUDA切换 一、查看当前PyTorch使用的CUDA版本: 注意

    2024年02月11日
    浏览(52)
  • Linux/Debian/Ubuntu-OpenCV(4.5.4/4.6.0)+CUDA(11.3)配置编译全流程

    本文用来记录在linux环境下docker中编译 OpenCV with cuda 的过程,同时编译了4.5.4和4.6.0两个版本均可编译通过。 本地是linux环境也可参考本文完成编译。 系统:debian 11 CPU:i7 内存:16G 显卡:Nvidia Quadro M2000 github直接下载连接: OpenCV 4.5.4 source code OpenCV 4.6.0 source code 如需下载其他

    2024年02月10日
    浏览(45)
  • Ubuntu 20.04(linux) cuda(12)+cudnn的deb方式安装以及验证(宝宝也适用哟)

    前言(碎碎念) 想当年在实验室就在自己电脑(双系统)和服务器上都搭建这个环境(Ubuntu18.04+conda+pycharm+cuda+cudnn完整流程) 还写了操作文档,主要遇到了太多问题(比如NVIDIA驱动安装后黑屏问题,真是想用linux之父的话来说一句) 现在再次遇到重装服务器这个事情,又遇

    2024年02月03日
    浏览(108)
  • Ubuntu18.04配置ZED_SDK 4.0, 安装Nvidia显卡驱动、cuda12.1

    首先卸载nvidia相关的、卸载cuda sudo apt-get purge nvidia* sudo apt-get autoremove sudo apt-get remove --auto remove nvidia-cuda-toolkit sudo apt-get purge nvidia-cuda-toolkit 官方卸载cuda的方法: 检查cuda还在不在: locate libcudart9.1 在软件更新中心里,手动安装显卡驱动。 ubuntu桌面-左下角九宫格-software upda

    2024年02月14日
    浏览(55)

觉得文章有用就打赏一下文章作者

支付宝扫一扫打赏

博客赞助

微信扫一扫打赏

请作者喝杯咖啡吧~博客赞助

支付宝扫一扫领取红包,优惠每天领

二维码1

领取红包

二维码2

领红包