PyTorch 编辑器的选择及配置

以下介绍的两种编辑器可二选一使用，也可以结合使用PyCharm

下载

• 进入官网：https://www.jetbrains.com/pycharm/
• 下载社区版（免费）

安装

• 设置安装路径（避免出现中文和空格）

• 根据自己的需要勾选
  • Create Desktop Edition：创建桌面快捷方式
  • Update PATH Variable (restart needed)：更新环境变量，可以在命令行中启动PyCharm
  • Update Context Menu：添加到菜单栏
  • Create Associations：关联所有以.py结尾的文件

• 其他选项默认

配置

• 打开PyCharm创建新项目

• 为新创建的项目设置存放路径

• 选择已配置的环境

• 选择Conda Environment，找到之前配置的环境地址

• 成功导入之间创建的PyTorch环境

检查环境是否成功导入

• 打开Python控制台

• 在控制台输入

import torch
torch.cuda.is_available()

• 出现True的字样说明导入成功

Jupyter

下载及安装

• 安装过Anaconda之后，Jupyter也会随着安装好，如果没有安装的话，看开发环境搭建
• 随Anaconda安装的Jupyter默认安装在（base）环境中，所以需要在自己创建的环境中安装Jupyter
• 打开Anaconda Prompt，进入之前创建的环境

• 输入conda install nb_conda安装Jupyter所需要的包

• 安装完成之后输入jupyter notebook

• 选择py38环境创建笔记

检查是否配置成功

• 同样输入以下代码

import torch
torch.cuda.is_available()

• 输入完一句之后按Shift+Enter，意思是执行上一句并跳转到下一句