1. 配置conda虚拟环境
(1)打开Anaconda Prompt
(2)输入命令
conda create -n pytorch python = 3.6
接着输入y,便可完成pytorch虚拟环境创建。
(3)进入Pytorch虚拟环境
输入:conda activate pytorch 进入到pytorch环境
如果能够顺利进入,则证明第一步安装成功。
2. 安装Pytorch
2.1 官网安装
(1)打开pytorch官网:https://pytorch.org/
上述配置主要需要注意的是系统、Pytorch版本。安装时,将最后一行的Command代码复制到conda命令行窗口运行即可
(2)Pytorch版本选择及注意事项
CUDA
如果选择安装CUDA版本需要注意:
- 电脑是否有英伟达(NVIDIA)的显卡。可以通过“右键 --> 任务管理器