pytorch 安装

背景：需提前安装cuda

查看支持的最大版本的cuda

nvidia-smi

这里我提前安装了11.8版本的

1、在这个网站查看对应的python版本和pytorch版本

添加链接描述
这里我的电脑cuda版本是11.8，对应版本为

即python版本为3.10，pytorch版本为2.1.0

2、在conda中创建Python环境：

conda create --name pytorch python=3.10

激活：

conda activate pytorch

查看所有环境：

conda env list

3、安装对应版本的pytorch

在pytorch官网查找对应版本
添加链接描述

用pip安装的指令：

pip install torch==2.1.0 torchvision==0.16.0 torchaudio==2.1.0 --index-url https://download.pytorch.org/whl/cu118

4、校验安装是否成功

输入指令导入发现报错了

输入指令降低numpy版本

pip install "numpy<2"

输入下面代码后出现版本信息则安装成功

import torch
print(torch.__version__)

后续学习用到的其他库：

pip install matplotlib
pip install IPython