一、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的包中,再下载