ubuntu最佳使用流程4：安装torch环境

1.查看cuda版本：

nvcc --version

2.新建环境

conda create -n yolo python = 3.8.0

onda activate yolo

3.安装torch

进入torch官网torch

寻找匹配版本

pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu121

# 进入python编译器
python

#进入python编译器,进一步输入
import torch

# 查看版本
torch.__version__

# 退出python
exit()

输出’2.2.1’等则安装正确