PyTorch学习一(环境配置与安装)

anaconda安装

官网:www.anaconda.com

anaconda一个集成的工具包

安装地址:https://repo.anaconda.com/archive/

选择Anaconda3-5.2.0-Windows-x86_64.exe版本(631.1M)

显卡作用:训练加速

显卡配置: 驱动 CUDA Toolkit

在任务管理器检查驱动安装:在性能部分查看GPU如果能正常显示型号则正常

img

小tips 如何管理环境

(conda create -n pytorch python=3.6)

img

他就像一个小房子在里面安装环境

使用conda 创建一个环境

(安装时全部y就行)

img

使用conda activate pytorch来激活

执行成功后可以看到左边括号里的名字变了,变成了新的我们激活的房间

使用pip list命令列出要使用的工具包

安装pytorch

官网地址:https://pytorch.org/

img

然后查看GPU型号

NVIDIA GeForce RTX 3070 Ti Laptop GPU

安装指令:

conda install pytorch torchvision torchaudio pytorch-cuda=12.1 -c pytorch -c nvidia

使用nvidia-smi命令查看驱动版本

img

不够就去官网升级(https://www.nvidia.com/Download/index.aspx?lang=cn

img

检测安装

输入python

输入import torch无报错则正常

输入torch.cuda.is_available()

返回true则正常(说明该计算机的GPU可以访问pytorch)