PyTorch环境的配置及安装

一、PyTorch环境的配置

一、安装anaconda

anaconda安装地址：www.anaconda.com

由于我的python使用的python3.6版本所以下载对应的anaconda版本
anaconda历史收录版本：repo.anaconda.com/archive/

这里选择的是anaconda3-5.2.0版本

安装完成后测试是否安装成功
点开anaconda prompt查看命令行中有无(base)

检查显卡的驱动是否正确安装
在任务管理器中点击性能，查看GPU是否能正常显示型号

使用conda指令创建PyTorch环境
输入conda create -n pytorch python=3.6
创建成功后会询问是否更新库，输入：y
最后使用：conda activate pytorch激活库
激活成功后左边的(base)会变成(pytorch)

![](https://img-blog.csdnimg.cn/c48dac9b4e1e473d81c23516ae1e349b.png#pic_center

二、安装PyTorch

进入PyTorch的官网：https://pytorch.org/
版本的选择：
PyTorch Build：Stable(1.3)
Package：windows下conda；Linux下pip
Language：python3.6
CUDA：在https://www.nvidia.cn/geforce/technologies/cuda/supported-gpus/?field_gpu_type_value=notebook-gpus中查看支持的GPU中是否有自己的电脑GPU，有则选择9.2，无则选择None
在命令行中输入：conda install pytorch=1.3 torchvision cudatoolkit=9.2 -c pytorch -c
选择是否安装：y
检验是否安装成功：
有些安装pytorch卡住的可以在网上找到这两个包的资源，放入
anaconda的pkgs的包中，再下载