pytorch安装教程

pytorch安装教程

本教程以Ubuntu为例部署pytorch，windows同理。

1、下载anaconda

pytorch可以理解为一个python库，类似于numpy等，因此可以使用anaconda进行管理。anaconda的下载可以直接去官方地址挑选对应的版本下载即可。下载好后利用sftp传到linux系统中，再到放置的文件夹下sh对应的安装文件即可，注意下载好后需要重启终端，conda指令才可以生效。

2、安装pytorch

到pytorch官方地址，我的选择版本如下：

推荐选择stable，os因为这里我是在Ubuntu中部署因此选择linux，package选择conda，语言选择python，硬件平台可以通过nvidia-smi指令查看cuda版本，我的cuda版本为11.4因此选择11.3，注意cuda版本不能选择比自己电脑的cuda版本高。

复制下方的指令并运行。

3、问题

出现网络问题报错如下

是因为conda源的问题，到清华官方源中复制以下内容

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
  pytorch-lts: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
  simpleitk: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud

替换/home下的自己用户名目录下的.condarc文件中的内容，指令如下：

vim /home/用户名/.condarc

之后便可以顺利安装。

4、验证

依次运行如下指令验证安装是否成功

python
import torch

若不报错就一般没问题，看cuda是否安装成功运行如下指令：

import torch
torch.cuda.is_available()

若输出为true则cuda版本安装成功。