Pytorch深度学习环境配置

始、Pytorch深度学习快速入门【小土堆】

版本:

Anaconda:Anaconda3-5.2.0-Windows-x86.64

Cuda:10.2

报错汇总:

1.python.exe无法找到入口

解决方法:找到报错路径下的pythoncom36.dll并删除


一、安装Anaconda

1.进入Anaconda历史版本页面:https://repo.anaconda.com/

2.选择下载Anaconda3-5.2.0-Windows-x86.64.exe

3.跳过安装vs


二、管理环境

1.创建环境:打开Anaconda Prompt,输入conda create -n pytorch python=3.6

2.激活环境:输入conda activate pytorch

3.查看环境包体:pip list


三、安装Pytorch

普通方法:

1.打开Pytorch首页:https://pytorch.org/

2.依次选择Stable1.9.1WindowsCondaPythonCUDA10.2,执行下方代码行:

1
conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch

清华源方法:

1.先添加清华镜像channel:

1
2
3
4
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
conda config --set show_channel_urls yes
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/

2.执行安装代码行:

1
conda install pytorch torchvision cudatoolkit=10.2

3.验证是否安装成功:

1
pip list

查看环境中是否有torch工具包


四、验证CUDA

1.按顺序输入以下命令:

1
2
3
python
import torch
torch.cuda.is_available()

如果返回True,则表示Pytorch可以正常使用GPU


五、编辑器

vscode

1.打开vscode官网下载:https://code.visualstudio.com/docs/?dv=win64user,安装包会自动下载

jupyter:

1.在pytorch环境中安装jupyter:打开Anaconda Prompt并进入pytorch环境然后输入conda install nb_conda

2.验证是否安装成功:

1
pip list

查看环境中是否有ipykernel工具包

3.启动jupyter:输入jupyter notebook