Pytorch安装太慢/安装不了怎么办?那就离线安装Pytorch

1、查看CUDA版本:命令nvcc --version

2、查看python3版本

3、下载相对应版本的Pytorch安装包

安装包下载地址:https://download.pytorch.org/whl/torch_stable.html

cpu对应与cpu版本,cu对于cuda版本

比如我的python版本是3.7,cuda版本是11.1

因此我下载的cuda版本是torch-1.8.1+cu111-cp37-cp37m-linux_x86_64.whl

cp37对应于python3.7,cu111对应于cuda版本11.1

4、安装Pytorch

首先执行:

pip install torchvision

然后进入安装包的路径,接着执行:

pip install torch-1.8.1+cu111-cp37-cp37m-linux_x86_64.whl

5、然后验证pytorch和torchvision是否安装成功,在python命令行输入import torch,无警告代表成功。

autopilot@autopilot-Default-string:~/下载$ python3
Python 3.7.11 (default, Jul 27 2021, 14:32:16)
[GCC 7.5.0] :: Anaconda, Inc. on linux
Type "help", "copyright", "credits" or "license" for more information.
>>> import torch
>>>

 至此,Pytorch安装成功!

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值