二、安装步骤1、操作系统选择查看pytorch官网,可以看到pytorch支持linux、mac、window平台、支持conda、pip、源码等安装方式,也支持cpu、cuda、rocm计算平台,我们点击环境选择可以发现目前只有linux系统是支持全语言、全安装方式、全计算平台的,所以我们选择linux操作系统作为系统环境。另外机器学习计算要求glibc版本要求较高,centos搭载的内核和glbic版本较低,ubuntu搭载的内核版本都较新,所以机器学习主机建议使用ubuntu操作系统。目前cuda更新支持的最低ubuntu版本为18.04,所以建议使用ubuntu18.04以上的操作系统。
wuhs@s169:~$ cat /etc/os-release
name=“ubuntu”
version=“18.04.6 lts (bionic beaver)”
2、anaconda3安装如上所示,pytorch支持的安装方式有多种,博主拟采用conda安装方式,建议先安装anaconda3,可以根据我们需要创建不同虚拟环境,虚拟环境下安装不同的机pytorch版本,虚拟环境支持互不影响。ubuntu环境下anaconda的安装见博文ubuntu之anaconda3安装。
wuhs@s169:~$ wget https://mirrors.bfsu.edu.cn/anaconda/archive/anaconda3-2022.10-linux-x86_64.sh
wuhs@s169:~$ sh anaconda3-2022.10-linux-x86_64.sh
wuhs@s169:~$ source ~/.bashrc
3、查看python版本不同的pytorch版本要求的python版本是不一样的,所以安装好anaconda3后我们检查当前的python版本,默认初始化都是当前anaconda3发布时对应的python最新版本,当然我们也可以使用conda创建所需的python环境版本。我们在torchvision查看pytorch、torchvision、python版本匹配要求。
(base) wuhs@s169:~$ python -v
python 3.9.13
4、安装pytorch如下第二步,pytorch官网我们可以在选择操作系统、安装方式、编程语言、计算平台后生成对应的安装命令。
(base) wuhs@s169:~$ conda install pytorch torchvision torchaudio cpuonly -c pytorch
collecting package metadata (current_repodata.json): done
solving environment: done
## package plan ##
proceed ([y]/n)? y
…
5、版本验证(base) wuhs@s169:~$ python
python 3.9.13 (main, aug 25 2022, 23:26:10)
[gcc 11.2.0] :: anaconda, inc. on linux
type “help”, “copyright”, “credits” or “license” for more information.
>>> import torch
>>> torch.version
‘1.13.1’
>>>
三、指定版本安装1、创建虚拟环境(base) wuhs@s169:~$ conda create -n pytorch python=3.9
…
(base) wuhs@s169:~$ conda activate pytorch
(pytorch) wuhs@s169:~$
2、安装指定版本的pytorch安装指定版本的pytorch的时候我们需要在github官网pytorch频道查看匹配版本,conda安装的时候指定版本号,具体版本号可以查看anaconda官网,pytorch版本和torchaudio对应关系见
torchaudio。当然如果我们指定版本错误的情况下,安装的时候会报错,我们根据报错提示核验是哪个软件版本指定错误,再去官网合适确认修正后重新安装即可。
(pytorch) wuhs@s169:~$ conda install pytorch2.12.0 torchvision=0.13.0 torchaudio0.12.0 cpuonly -c pytorch
以上就是linux下pytorch安装的方法是什么的详细内容。
