pytorch下载太慢怎么办?

刚刚阅读1回复0
kewenda
kewenda
  • 管理员
  • 注册排名1
  • 经验值156875
  • 级别管理员
  • 主题31375
  • 回复0
楼主

windows11 cuda11.3 torch1.10.0 离线安拆。

1.在pytorch官网(Previous PyTorch Versions | PyTorch)选择本身想要的版本(本文选择1.10.0),能够看到对应的安拆代码如下,在此是为了查看需要安拆包的版本:

# CUDA 11.3 conda install pytorch==1.10.0 torchvision==0.11.0 torchaudio==0.10.0 cudatoolkit=11.3 -c pytorch -c conda-forge

2.在 https://download.pytorch.org/whl/torch_stable.html 找到对应版本的安拆包,cu113即cuda11.3版本,torch1.10.1是torch版本(与官网给的下载代码版底细同),cp38即python3.8,win为Windows系统。

别离找到torchvision,torchaudio寻找办法与上面不异。

cudatoolkit能够间接在官网下载(下载差别班的的间接搜刮:cudatoolkit+版本号,若是官网进不去,能够把com改成cn)

torch,torchvision,torchaudio下载完成后如图所示,

3.找到本身创建的conda情况,我那里新建了一个名为pytorch的情况,至于若何新建能够查看:

巫婆:Anaconda-用conda创建python虚拟情况363 附和 · 19 评论文章

4.将当地包导入到conda中的pytorch情况中,进入anaconda主页面,找到对应的pytorch情况,点击 Open Terminal进入号令行窗口。

别离输入如下代码:

pip install D:\Anaconda\离线包\torch-1.10.0+cu113-cp38-cp38-win_amd64.whl D:\Anaconda\离线包替代成你存放三个包的位置 pip install D:\Anaconda\离线包\torchaudio-0.10.0+cu113-cp38-cp38-win_amd64.whl pip install D:\Anaconda\离线包\torchvision-0.11.0+cu113-cp38-cp38-win_amd64.whl

我在安拆torchvision时,可能那个包需要相关依赖包,我那里在anaconda中pytorch中先安拆了pillow,numpy包(间接停止pip也能够安拆,它会主动帮你安拆相关依赖,但是我那里都失败了,所以我在情况下本身安拆了相关包)

5.验证,我那里利用的是pycharm,新建工程-选择已经设置装备摆设好的解释器(我那里为anaconda中新建的pytorch)

若是没有的话选择添加解释器-conda

在main.py中输入如下代码:

import torch # 能否挪用pytorch库 print(torch.cuda.current_device()) # 输出当前设备(我只要一个GPU为0) print(torch.cuda.device(0)) # <torch.cuda.device object at 0x7fdfb60aa588> print(torch.cuda.device_count()) # 输出含有的GPU数目 print(torch.cuda.get_device_name(0)) # 输出GPU名称 我那边为:NVIDIA GeForce RTX 3060 Laptop GPU x = torch.rand(5, 3) print(x) # 输出一个5 x 3 的tenor(张量)

输出:

0 <torch.cuda.device object at 0x000001ECF20AADC0> 1 NVIDIA GeForce RTX 3060 Laptop GPU tensor([[0.6157, 0.2930, 0.2095], [0.4757, 0.5016, 0.8916], [0.9307, 0.7330, 0.9883], [0.1007, 0.6968, 0.1915], [0.7748, 0.6465, 0.4031]])

0
回帖 返回游戏

pytorch下载太慢怎么办? 期待您的回复!

取消
载入表情清单……
载入颜色清单……
插入网络图片

取消确定

图片上传中
编辑器信息
提示信息