PyTorch 编辑器的选择及配置

2021-08-27 11:16:28 浏览数 (1)

以下介绍的两种编辑器可二选一使用,也可以结合使用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控制台
  • 在控制台输入
代码语言:python代码运行次数:0复制
import torch
torch.cuda.is_available()
  • 出现True的字样说明导入成功

Jupyter

下载及安装

  • 安装过Anaconda之后,Jupyter也会随着安装好,如果没有安装的话,看开发环境搭建
  • Anaconda安装的Jupyter默认安装在(base)环境中,所以需要在自己创建的环境中安装Jupyter
  • 打开Anaconda Prompt,进入之前创建的环境
  • 输入conda install nb_conda安装Jupyter所需要的包
  • 安装完成之后输入jupyter notebook
  • 选择py38环境创建笔记

检查是否配置成功

  • 同样输入以下代码
代码语言:python代码运行次数:0复制
import torch
torch.cuda.is_available()
  • 输入完一句之后按Shift Enter,意思是执行上一句并跳转到下一句

0 人点赞