PyTorch环境的配置及安装

1、安装Anaconda
https://repo.anaconda.com/
安装相应的版本,我这里安装的是Anaconda3-5.2.0-Windows-x86_64.exe

PyTorch环境的配置及安装_第1张图片
PyTorch环境的配置及安装_第2张图片
PyTorch环境的配置及安装_第3张图片
PyTorch环境的配置及安装_第4张图片

2、显卡准备
深度学习离不开显卡(PyTorch支持英伟达显卡)
检查显卡驱动是否正常安装:正常显示表示,驱动正常安装。
PyTorch环境的配置及安装_第5张图片

3、管理环境

3.1、
PyTorch环境的配置及安装_第6张图片
PyTorch环境的配置及安装_第7张图片

3.2、初始环境(base) (初始的屋子)
下面我们创建另一个环境(pytorch):

conda create -n pytorch python=3.6

PyTorch环境的配置及安装_第8张图片

3.3、激活环境(pytorch):

conda activate pytorch

PyTorch环境的配置及安装_第9张图片

3.4、查看这个环境(屋子)中有哪些包

pip list

PyTorch环境的配置及安装_第10张图片
很明显,上面的包中没有pytorch包,所以我们要安装pytorch

3.5.安装pytorch
3.5.1、进入pytorch官网:https://pytorch.org/
英伟达显卡,如下操作:
PyTorch环境的配置及安装_第11张图片

选好后,复制运行指令:

conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch

3.5.2、查看驱动版本是否符合conda要求
命令窗口 输入nvidia-smi,查看驱动版本。
conda仅支持驱动版本号大于396.26
PyTorch环境的配置及安装_第12张图片

3.5.3,、
在环境(pytorch)中运行3.5.1复制的指令

conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch

PyTorch环境的配置及安装_第13张图片
再查看一下安装的包有哪些

pip list

PyTorch环境的配置及安装_第14张图片

很明显,pytorch包成功安装

3.6、检查pytorch是否安装成功
在pytorch环境中 输入 python
输入import torch若没有报错,正常显示,则pytorch成功安装。
PyTorch环境的配置及安装_第15张图片

3.7、检查pytorch是否可用使用你电脑的GPU
输入torch.cuda.is_available()
如果显示True,则你的这台计算机的GPU可以给pytorch使用
PyTorch环境的配置及安装_第16张图片

至此,大功告成。

你可能感兴趣的:(pytorch,深度学习,python)