PyTorch的简介和安装

本文主要参考：datawhale的学习材料
Pytorch简介与安装

学习目标：

1、安装pytorch 和 pycharm
2、配置相关环境

学习内容：

1.Anaconda的安装
2.检查NVIDIA GPU
3.安装PyTorch

学习笔记：

1.安装anaconda

1.1 按照官网的步骤安装anaconda
1.2 创建虚拟环境：

a.查看本机已有环境，在Anaconda prompt 中输入

conda env list ;

本机仅有一个环境

b.创建虚拟环境learn ，在Anaconda prompt 中输入

conda create -n learn python=3.8.8

c.Anaconda换源
在安装package时，我们经常会使用pip install …和conda install … 的命令，但是一些package下载速度会很慢。

pip 换源
1、文件管理器文件路径地址栏敲：%APPDATA% 回车，快速进入 C:Users电脑用户AppDataRoaming 文件夹中。
2、新建 pip 文件夹并在文件夹中新建 pip.ini 配置文件。
3、我们需要在pip.ini 配置文件内容，你可以选择使用记事本打开，输入以下内容，输入完后记得按下ctrl+s保存哦，在这里我们使用的是豆瓣源。

通过pip安装pandas包，发现下载地址已经变成pypi.douban.com/simple

conda 换源

TUNA 提供了 Anaconda 仓库与第三方源的镜像，各系统都可以通过修改用户目录下的 .condarc 文件。
Windows 用户无法直接创建名为 .condarc 的文件，可先执行

conda config --set show_channel_urls yes

生成该文件之后再修改。

完成这一步后，我们需要在C:Users电脑用户找到 .condarc这个文件,打开后将文件里原始内容删除，将下面的内容复制进去，并且保存。

channels:
  - defaults
show_channel_urls: true
default_channels:
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/r
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/msys2
custom_channels:
  conda-forge: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
  msys2: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
  bioconda: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
  menpo: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
  pytorch: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
  simpleitk: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud

这一步完成后，我们需要打开Anaconda prompt 运行

conda clean -i

清除索引缓存，保证用的是镜像站提供的索引。

2.任务管理器查看自己的显卡

3.安装pytorch

a. 根据本机的情况选择安装

conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch

ps：将pytorch 安装在之前创建的虚拟环境里

b.检验安装成功与否

import torch
torch.cuda.is_available()

显示已安装成功。

PyTorch的简介和安装

Python相关栏目本月热门文章