Pytorch虚拟环境配置

619 阅读3分钟

参见b站【PyTorch深度学习快速入门教程(绝对通俗易懂!)【小土堆】】www.bilibili.com/video/BV1hE…

安装Anaconda

www.anaconda.com/ (官网傻瓜式安装即可)

打开控制台,可以看到【base】即表示安装成功

image.png

Anaconda常见指令

查看已创建的虚拟环境:conda env list
激活已创建的虚拟环境:conda activate 环境名
退出激活的虚拟环境:conda deactivate
删除一个已有的虚拟环境:conda env remove --name 环境名 
创建一个新的虚拟环境:conda create -n 环境名 python=python版本号
查看当前环境安装的包:conda list 或 pip list

虚拟环境创建与Pytorch下载配置

为什么需要创建虚拟环境?

在安装某版本PyTorch库的过程中,会自动替换其所有依赖库(比如numpy等科学计算基础库)至相应匹配的版本。而这些基础库又是其他某些高级库(比如Tensorflow、PaddlePaddle)的重要依赖,它们之间也有密切的版本对应关系;又由于那些高级库与PyTorch无关,所以在安装PyTorch的时候不会去自动替换掉那些高级库的版本。因此,自动换掉的基础库很有可能与其他高级库产生不匹配的冲突,导致原先功能无法正常使用,也就是原先的运行环境被污染了。为了避免这种污染的发生,最好新建一个环境,单独安装PyTorch及其依赖的所有库,互不干扰,避免毁坏原先辛辛苦苦配好的运行环境。

  1. 创建虚拟环境,这里将虚拟环境名命名为 pytorch ,后续都用该名指代虚拟环境名
python --version #查看当前python环境,笔者python版本为3.9.13
# 语法:conda create -n 环境名 python=python版本号
conda create -n pytorch python=3.9.13 #创建一个虚拟环境,命名为pytorch。
  1. 在虚拟环境中安装 Pytorch

进入Pytorch官网 pytorch.org 选择对应自己电脑的参数

注:选择不同参数对应的指令不同。这里以自己电脑的为主 image.png

⚠️ 首先需要切换到自己的虚拟环境,如这里是【conda activate pytorch】,再复制该指令进入虚拟环境进行安装。

image.png

Jupyter notebook 中使用 Pytorch

安装完 Anaconda 后 jupyter notebook 已经自动下载,但是默认存放在 base 环境下,无法使用 Pytorch (因为我们刚才将 Pytorch 安装在新环境 pytorch 中)。因此,要在 jupyter notebook 中使用 Pytorch ,有两种解决方案。

  1. 在 base 环境安装 Pytorch
  2. 在 pytorch 环境配置 jupyter notebook

这里采用第二种方式,解决方案如下:

  1. 确保所在虚拟环境为 pytorch
conda activate pytorch
  1. 下载 Jupyter notebook
conda install nb_conda_kernels #python3.9以上
conda install nb_conda #python3.9以下
  1. 查看下载是否成功
jupyter notebook #控制台打开 Jupyter notebook
  1. 配置 Jupyter notebook

image.png

image.png

Pycharm 中使用 Pytorch

image.png

Interpreter地址根据自己pytorch下载路径进行选择,位于/bin/python

image.png

打开Pycharm命令控制台(Python Console)进行验证

image.png