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