您当前的位置:首页 > 计算机 > 编程开发 > Python

win10下安装pytorch,torchvision

时间:03-12来源:作者:点击数:

电脑里以前安装了 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 即可解决!

方便获取更多学习、工作、生活信息请关注本站微信公众号城东书院 微信服务号城东书院 微信订阅号
推荐内容
相关内容
栏目更新
栏目热门