本方法用于无法用官网的方法安装,主要参考这位大佬的博客离线安装Pytorch 最简单 高效的方法_pytorch离线安装_正在学习的浅语的博客-CSDN博客

 1、创建环境

首先要用conda创建一个虚拟环境,到时候把pytorch装到这个环境里,详细如何创建就不多介绍了

2、下载地址

官网下载,这个网址是官方的离线安装包

3、查找对应的版本

点开网址后,我们可以看到一系列的安装包,如下图所示

 这里讲一下如何找到自己所需要的安装包

首先是解释一下这些包的各个部位的含义,首先是最前面的,cpu或者cu110等,这些代表着这个是cpu还是gpu的,而cu110则代表gpu的cuda版本为11.0。cp代表Python版本,如cp36则代表Python3.6版本。最后的win_amd和linux_x比较容易理解,代表系统版本。

举个例子,如果我的电脑是winodws,虚拟环境的Python是3.9版本,像安装GPU版的Pytorch,cuda版本想用11.6,则我应该下载这个cu116/torch-1.13.1%2Bcu116-cp39-cp39-win_amd64.whl

 同理,把其他的都下载,如torchaudio、torchvision等对应版本都下载下来。

 4、本地安装

1、先在你下载文件的路径,用cmd打开命令管理窗口,然后输入以下代码:

conda activate 环境名

2、在激活你的环境后,输入下列代码,在该环境下安装Pytorch

pip install 文件名.whl

 上面代码中的文件名换成你下载的安装包名字,依次安装你下载的本地安装包。

3、在安装完毕后,在环境中打开Python,并在Python中输入下列代码以验证是否安装成功

import torch
print(torch.__version__)

如果可以正常运行,就说明成功安装了

离线安装Pytorch 最简单 高效的方法_pytorch离线安装_正在学习的浅语的博客-CSDN博客

Logo

有“AI”的1024 = 2048,欢迎大家加入2048 AI社区

更多推荐