win10下安装pytorch,torchvision

电脑里以前安装了 tensorflow,现在因为学习需要,需要安装pytorch。还是在原来安装tensorflow的位置安装pytorch。

由于采用在线安装太慢了,而且中途还会因为网速不稳定终端!因此,采用离线安装(真的是快啊),先将具体操作记录如下

(PS:本文只讲在已经安装CUDA和CUDNN等一系列前提下,进行的)

1、首先去清华镜像下载pytorch

https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/win-64/

找到对应的版本下载即可,本人安装的是

2、打开anaconda prompt,执行如下操作

(base) D:\software\Anaconda3>conda install --offline -n base pytorch-nightly-1.0.0.dev20190101-py3.6_cuda90_cudnn7_0.tar.bz2

(注意 base是默认的虚拟环境,回车等待安装即可)

3、安装pytorch的支持库 torchvision

pip3 install --no-deps torchvision

4、测试

注意:  在安装torchvision的时候遇到的错误

解决办法,将pip install torchvision  改为 pip3 install --no-deps torchvision  即可解决!

猜你喜欢

转载自www.cnblogs.com/guohaoblog/p/10340360.html