pip 安装pyTorch

pip install torch1.3.0+cu92 torchvision0.4.1+cu92 -f https://download.pytorch.org/whl/torch_stable.html --no-cache-dir

安装的torch版本, cuda版本, torchvision版本 需要对应
官网地址

pip库安装临时源: -i https://pypi.tuna.tsinghua.edu.cn/simple

测试gpu

import torch
torch.cuda.is_available()
# 输出True则安装成功
Logo

北京人形旗下天工造物具身智能开源社区,聚焦具身天工与慧思开物两大平台

更多推荐