Ubuntu20.04安装torch

描述

根据python版本安装对应的pytorch

安装

首先查询当前环境的python版本

conda activate base
python

可以看出我的python版本是3.8

Python 3.8.13 (default, Mar 28 2022, 11:38:47) 
[GCC 7.5.0] :: Anaconda, Inc. on linux
Type "help", "copyright", "credits" or "license" for more information.
>>>

接下来去pytorch的官方下载页
https://download.pytorch.org/whl/cpu/torch_stable.html

选择适配当前python版本的.whl文件下载

我选择的是torch-1.12.1+cpu-cp38-cp38-linux_x86_64.whl

选择事项：由于我们是Ubuntu系统，所以后缀首先是linux_x86_64.whl为结尾的。前一项cp38-cp38，适配你的python版本。注意第一项选择torch

下载之后，使用pip进行安装

cd ~/Downloads
pip3 install torch-1.12.1+cpu-cp38-cp38-linux_x86_64.whl

确认安装成功

终端运行

python

代码来返回版本号

import torch
print(torch.__version__)