pytorch环境配置——Anaconda3+pytorch+pycharm+Jupyter

pytorch环境配置（Anaconda3+pytorch+pycharm+Jupyter）

最近在做深度学习方面的东西，在配置pytorch环境的时候遇到了许多坑，现在将最完整的步骤分享给大家，帮助大家少走弯路。

下载安装Anaconda

Anaconda是一个集成了应用于科学分析的大量package的软件。
https://www.anaconda.com

点击Download自动下载 python3.10.9

注意

红色要√，不然后面要单独配置环境变量
skip 跳过安装VSCode

检验是否安装成功

能看到（base）则意味着安装成功

管理环境
目的：实现多python环境切换
先查询一下自己安装的python版本
cmd->python 我的是v3.10.9

然后打开Anaconda Promopt 输入

conda create --name pytorch python=3.10.9

创建一个新的pytorch环境
一直y就行
激活pytorch环境

conda activate pytorch

PyTorch的安装

https://pytorch.org/
打开主页，下滑↓

CUDA只能安装在NVIDIA的显卡上面，所以在安装之前，先确保自己有NVIDIA的显卡，并且已经成功安装驱动

查询自己的显卡驱动是否成功安装
打开任务管理器->性能->GPU
在右上角显示GPU型号，则成功安装显卡驱动

之后还需要知道自己CUDA安装的版本
命令行输入

nvidia-smi

我的CUDA支持的版本是11.6，由于显示的版本没有11.6，我就直接把最后安装的command的最后从cu117，改为了cu116
把该命令输入到之前新建的pytorch环境中->回车

pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu116

等待安装y即可

检验CUDA是否安装成功
命令行输入

nvcc -V

能显示对应版本即为安装成功

检验安装

在pytorch环境下输入

python
import torch

如果没有报错，则说明pytorch安装成功了

检验pytorch是否能使用GPU
继续输入

torch.cuda.is_available()

显示true则为安装成功

Python编辑器的选择及安装

PyCharm的安装

https://www.jetbrains.com/pycharm/

官网首页，点击DownLoad
选择社区版下载即可

使用PyCharm配置环境

打开PyCharm，创建一个新的项目
展开编译器的选择，选择现有的编译器
点击Add Interpreter
选择anaconda3\envs\PyTorch\python.exe
点击create

检验PyCharm是否成功导入PyTorch环境

打开Python Console
输入命令

import torch
torch.cuda.is_available()

返回true即为安装成功

Jupyter的安装

在pytorch环境下安装Jupyter
打开Anaconda Prompt

输入

conda install nb_conda

自动安装，选择y即可

安装完成后输入

jupyter notebook

页面跳转
点击NEW->Python3(ipykernel)

验证该环境中是否含有pytorch
输入之前的指令

import torch
torch.cuda.is_available()

返回true则成功