Ubuntu安装配置Cuda和Pytorch gpu

本文档详细介绍了如何在Ubuntu操作系统中安装Cuda和Pytorch GPU版本。首先,根据Ubuntu版本选择合适的Cuda版本进行下载安装,并验证安装成功。接着,提供了如何在Pytorch官网获取最新或特定版本的Pytorch,并通过conda或pip命令安装到虚拟环境中,以确保版本匹配。
摘要由CSDN通过智能技术生成

前言

在Ubuntu中操作系统中,通过Anconda安装对应的虚拟环境以及软件包,一般都需要适配Cuda、Pytorch版本等

以下安装配置都是在Ubuntu操作系统下

1. 安装Cuda

通过Ubuntu操作系统查看cuda适配的版本:nvidia-smi
截图如下:
在这里插入图片描述

查看Ubuntu版本可如下方式

  • (1)cat /proc/version
  • (2)uname -a
  • (3)lsb_release -a

在官方选择适配的版本:Cuda官网,并选择相应的配置下载:
在这里插入图片描述
检查是否安装成功:nvcc -V

2. 安装Pytorch

项目文件中不需要pytorch的规定版本,可以在官方网站中选择最新即可:Pytorch最新版本<

  • 2
    点赞
  • 3
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 3
    评论
评论 3
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

码农研究僧

你的鼓励将是我创作的动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值