python环境配置步骤二:windows中Anaconda创建虚拟环境安装pytorch并在pycharm中使用虚拟环境

作者介绍

张伟伟,男,西安工程大学电子信息学院,2019级硕士研究生,张宏伟人工智能课题组。
研究方向:机器视觉与人工智能。
微信公众号:可随时查阅,搜索—张二牛的笔记,回复“python‘’获取整理好的专栏。
电子邮件:zhangweiweicpp@163.com

安装前先下载安装包,建议下载博主已准备好百度网盘链接:

百度网盘链接: https://pan.baidu.com/s/1FEv7L3YFN12lNdUWDwu6wQ
提取码:2021

windows中Anaconda创建虚拟环境安装pytorch

注:本教程安装版本为pytorch=1.1.0,对应cuda=9.0,torchvision=0.3.0,python=3.6,预计耗时30分钟可完成,查找显卡驱动可能需要些时间
安装前必备条件:
查询自己电脑是否支持GPU运算.
(1)若要安装GPU版本的pytorch,则需要满足显卡驱动的要求,不注意显卡驱动的要求安装GPU后在本教程第6步验证时可能会出现False。本教程尽量满足大家显卡驱动的版本,选择常用的cuda9.0,对应的显卡驱动版本需>=385.54,否则安装的版本不能使用GPU运算。如下图为不同的驱动版本所支持的cuda(cuda为加速运算的引擎),驱动版本向下兼容。
在这里插入图片描述
查询自己的显卡驱动所支持的cuda版本.查询到驱动版本大于385.54即可,若不满足,需要更新驱动或者安装cpu版本的pytorch。(如下图我所使用是驱动精灵软件、或者使用官方的GeForce Experience)
在这里插入图片描述
(2)安装cpu版本的pytoch
安装cpu版本的pytoch在本教程的第3步运行完之后,进入了虚拟环境,使用pytoch官网指定的方式安装即运行如下指令:

conda install pytorch-cpu==1.1.0 torchvision-cpu==0.3.0 cpuonly -c pytorch

下面开始我们的pytorch快速安装之旅吧!

  • 键盘使用win+R快捷键
    在这里插入图片描述
  • 输入cmd点击确定进入windows命令行终端
    在这里插入图片描述
    安装包时需要比较快的镜像源,因此选择清华源。

1.添加镜像源

在命令行依次运行下列5条指令添加。

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge 
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/msys2/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
conda config --set show_channel_urls yes

2.创建虚拟环境

创建虚拟环境(例如安装的虚拟环境名为py36,python版本指定为3.6),终端运行如下指令:

conda create -n py36 python=3.6

此时打开Anconda点击Environments发现有了名为py36的虚拟环境。
在这里插入图片描述
接下来就可以在名为py36这个虚拟环境中安装各种科学计算的包和框架(pytorch),若有多个环境,那么例如需要python3.7,那么就创建python3.7的虚拟环境,有conda管理不会出现版本冲突。
进入和退出虚拟环境使用指令如下图(图中为进入虚拟环境名为py36,并且验证python是否安装成功)

3.进入虚拟环境

终端运行如下指令为进入虚拟环境

conda activate py36

另外,若要退出环境则在看终端输入:

conda deactivate

进入虚拟环境如图:会有小括号(虚拟环境名:py36),进入虚拟环境后就可以在python为3.6的环境下配置pytorch了,尽情安装吧!
在这里插入图片描述

4 在虚拟环境中安装pytorch

本教程利用终端指令源码编译安装,最为快速,因此windows与Ubuntu系统类似。接下来安装pytoch1.1.0torchvision0.3.0

  • 下载pytoch1.1.0的源码包的网站:(安装包已经下载好并提供给大家,在文件中找到如下图片命名的安装包即可,以后若需要安装别的版本的去此网站下载,同理安装
    https://download.pytorch.org/whl/cu90/torch_stable.html
    选择并下载:
    在这里插入图片描述
    我所创建的虚拟环境起名为py36six,进入虚拟环境后,找到安装源码包的位置,例如我的路径在:E:\CDESKTOP\安装教程\torch-1.1.0-cp36-cp36m-win_amd64.whl
    在这里插入图片描述
    利用pip运行如下指令安装,很快便可编译,不会因为网络问题等出现任何错误,这至此,pytoch1.1.0安装完成,还需要安装torchvision:
pip install E:\CDESKTOP\安装教程\torch-1.1.0-cp36-cp36m-win_amd64.whl

在这里插入图片描述

5 在虚拟环境中安装torchvision

pytoch和torchvision和python是有对应的版本关系的,本教程查到pytoch1.1.0对应的torchvision版本为0.3.0。补充,pytoch和torchvision的版本对应关系可在 pytorch官网查询。对应关系也可参考下图:
在这里插入图片描述
下面使用pip指令安装torchvision0.3.0,参考链接.:

pip install --no-deps torchvision==0.3.0

在这里插入图片描述

6 验证是否安装完成

安装完成后,在虚拟环境中依次运行如下3指令进行验证pytoch是否安装完成和GPU是否可用

python 
import torch 
torch.cuda.is_available() 

在这里插入图片描述
import torch 不报错误说明torch安装成功,cpu版本的验证本条指令成功即可。
torch.cuda.is_available() 结果显示True,那么恭喜你GPU版本安装完成。

pycharm中使用虚拟环境

操作步骤可参考这篇简洁明了的博客:
手把手教你如何在Pycharm中加载和使用虚拟环境.熟悉pycharm的界面和运行调试等快捷键有必要自行学习。

参考链接:

课题组踩坑总结经验,希望大家不吝一键三连,关注支持一下!!!

  • 2
    点赞
  • 14
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值