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版本安装成功。