Anaconda下pytorch环境的创建,pytorch,torchvision的安装(cpu)

1.首先创建一个pytorch的环境,

先打开Anaconda Prompt对话框,然后直接输入

conda create -n pytorch python==3.6#这里指定了python的版本,如果不指定则默认创建的最新版本的python

然后激活创建的pytorch环境,在pytorch环境中安装torch和torchvision

conda activate pytorch#之后进入到pytorch环境

在安装torch的时候,直接pip install安装会很慢很慢,这里建议先下载要安装的torch版本,然后再用pip install (下载的torch文件位置)

pip install D:/ProgramFiles/Anaconda/envs/torch-1.1.0-cp36-cp36m-win_amd64.whl#安装torch

此时注意在安装torch的同时会安装numpy,如果numpy安装出现错误则会安装不成功。

检查torch安装是否成功,则在pytorch环境下输入python进入命令行模式,然后输入import torch如果没有错误则证明安装成功。

接下来退出命(exit())令行的模式回到pytorch的环境中安装torchvision,torchvision的安装和torch类似,先下载自己需要安装的版本,然后再在对应的目录进行安装。

torchvision的下载:

https://pypi.org/project/torchvision/0.1.8/

点击网址中左侧的"Release history",可以下载自己想要的版本,注意torchvision的版本对torch的版本又要求。

在安装torchvision的时候会要求安装pillow的版本,在这里我安装的torchvision是0.3.0版本的要求pillow版本大于等于4.1.1,当时我直接pip install pillow的时候默认安装的是pillow的最新版本。如果安装最新版本则在torchvision安装成功后,在测试环境中输入import torchvision会出错,显示找不到此模块。这是因为pillow的高版本和torchvision要求的4.1.1不匹配,因此需要用pillow4.1.1版本,若此时直接再安装pillow4.1.1版本还是不行。因此需要将torchvison卸载后先安装pillow4.1.1才可行。所以最好的方式是再安装torchvision0.3.0之前先安装pillow4.1.1。

pip install pillow==4.1.1#安装pillow
pip install D:/ProgramFiles/Anaconda/envs/torchvision-0.3.0-cp36-cp36m-win_amd64.whl
#安装torchvision0.3.0版本

如下图所示即为安装成功。

评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值