win10下安装pytorch,torchvision

 

电脑里之前安装了 tensorflow,如今由于学习须要,须要安装pytorch。仍是在原来安装tensorflow的位置安装pytorch。学习

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

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

 

一、首先去清华镜像下载pytorchip

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

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

 

二、打开anaconda prompt,执行以下操做tensorflow

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

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

 

三、安装pytorch的支持库 torchvisionim

pip3 install --no-deps torchvision

 

四、测试

 

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

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

相关文章
相关标签/搜索