Pytorch环境的配置及安装

Pytorch环境的配置及安装

管理环境

遇到不同的项目，需要使用不同版本的环境，比如该项目要用到Pytorch 0.4，另一个项目要用到Pytorch 1.0，如果卸载了0.4版本，安装了1.0版本，那下一次，再碰到0.4版本，就需要卸载1.0版本，安装0.4版本。

Anaconda集成的 conda 包就能解决这个问题，即造出两个屋子，相互隔离。一个屋子放 0.4 版本，一个屋子放 1.0 版本。需要哪个版本，就进哪个屋子工作。

首先使用 conda 指令创建一个屋子，叫做Pytorch

指令如下：

conda create -n pytorch python=3.6

done
#
# To activate this environment, use
#
#     $ conda activate pytorch
#
# To deactivate an active environment, use
#
#     $ conda deactivate

检验是否下载好

pip list

python
import torch   #未报错即可
torch.cuda.is_available()   #检查pytorch是否可以使用GPU，返回True即可