描述
根据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__)