linux服务器安装gpu版本的pytorch

每次换个环境又要安装pytorch,最开始安装总是出错,写下不踩坑安装全过程

1、首先看一下cuda的版本

运行命令 nvidia-smi

如下所示

 比方说我的是右上角显示11.2,所以

2、去PyTorch 官网

 选择Linux ,这里没有我的cuda版本

 

 找过往版本Previous PyTorch Versions | PyTorch

 发现有11.3  11.1版本和10.2版本,但是没有11.2版本,查阅得知可以向下兼容安装11.1版本的

 3、往下滑找到cuda对应版本(11.1)版本的torch

 按照以下命令安装

pip install torch==1.10.1+cu111 torchvision==0.11.2+cu111 torchaudio==0.10.1 -f https://download.pytorch.org/whl/cu111/torch_stable.html

等待……

安装成功

4、检查是否安装成功

以及确定是不是gpu版本的

 输入python回车

返回true代表安装gpu版本的torch成功

  • 8
    点赞
  • 10
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值