Pytorch学习笔记——环境配置安装

1、下载和配置环境

Anacodna必备（工具包里面都包含的有，集成与运用科学分析的软件，比较方便）

点击这个网页:Download Now | Anacondahttps://www.anaconda.com/download/success

按照教程安装（教程可以自己在网上搜），安装完成之后，点击所示位置，出现(base)..表示安装成功

2、英伟达显卡配置

显卡的作用是：训练加速

打开任务管理器，即可看到显卡型号

再打开显卡控制面板

即可查看显卡支持CUDA的版本

查看显卡CUDA版本号信息

3、管理项目环境

后期面对不同项目，可能项目的配置环境都不一样，需要管理项目环境

新版本需要3.8以上，应装3.8版本及以上

conda create -n pytorch python=3.8 完成后，

激活环境 conda activate pytorch

查看安装列表 pip list

后续的操作基于新建的pytorch的环境

4、安装pytorch

官网：PyTorchhttps://pytorch.org/

选择界面：

11.8和12.1都可以选，看个人需求以及cuda版本

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

如果不行，用pip命令

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

不要用清华镜像，否则下的是cpu版本的

5、验证是否安装成功

import torch torch.cuda.is_available()

若返回True，则安装成功

6、Jupyter在虚拟环境配置

conda install nb_conda_kernels

运行jupyter，命令为

jupyter notebook

成功后如图所示