Pytorch离线安装教程

我们总共需要4个文件:

image-20200811033205404

下载地址

链接:https://pan.baidu.com/s/1cUMFYLF3CVfHmXc1hUZrbA
提取码:nqoa

步骤一:安装两个包

下载地址:http://download.pytorch.org/whl/torch_stable.html

打开anaconda prompt

列出环境

conda env list

image-20200811012228852

删除环境

conda env remove -n (你要删的环境)

新建环境

conda create -n pytorch16 python=3.8

激活环境

conda activate pytorch16

找到安装包存放的路径

image-20200811033311259

通过cd命令去到存放包的目录下

image-20200811033810405

在安装离线包的时候同时会下载相关的包,我们添加镜像源加快下载速度:

1
pip config set global.index-url https://mirrors.aliyun.com/pypi/simple

然后执行下面命令:

1
pip install torch-1.6.0+cu101-cp38-cp38-win_amd64.whl

然后继续安装第二个包:

1
pip install torchvision-0.7.0+cu101-cp38-cp38-win_amd64.whl

步骤二:安装显卡驱动

下载地址

https://developer.nvidia.com/cuda-10.1-download-archive-base

image-20200811021406497

如果之前安装过cuda要先卸载,卸载方法如下:

image-20200811022215583

然后删除文件夹:

image-20200811022324493

下载好之后安装(全部默认下一步)

下载cudnn:

https://developer.nvidia.com/rdp/cudnn-download

image-20200811023056355 image-20200811023149910

然后解压cudnn,将里面的内容复制到cuda文件夹

image-20200811024541974

步骤三:验证安装成功

1
2
3
4
import torch

print(torch.__version__)
print(torch.cuda.is_available())

输出

1
2
3
4
1.6.0+cu101
True

Process finished with exit code 0

成功