Linux pytorch + cuda 搭建gpu训练环境

本文介绍了如何使用pip自动安装PyTorch,包括检查服务器CUDA版本并根据其选择对应版本的安装方法。同时,针对网络不可用的情况,也提供了手动下载并使用conda或pip在本地安装的步骤,以适应不同环境需求。
摘要由CSDN通过智能技术生成

一、使用pip自动安装:

1. 使用 nvidia-smi 命令查看服务器对应的cuda版本及驱动版本,如下图所示:

2.根据对应cuda的版本,去pytorch官网Previous PyTorch Versions | PyTorch

,按照指定命令进行安装【测试发现使用pip命令能够成功安装】:

pip install torch==2.2.0 torchvision==0.17.0 torchaudio==2.2.0 --index-url https://download.pytorch.org/whl/cu121

二、手动下载安装:

如果有服务器无法联网,只能手动下载的情况,可以先到对应的镜像源(如清华镜像源)下载对应的torch+cuda+cudnn,然后放到本地,使用conda或者pip命令进行安装

  • 4
    点赞
  • 5
    收藏
    觉得还不错? 一键收藏
  • 1
    评论
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值