ubuntu20.04快速安装pytorch（cpu和gpu版本）

最新推荐文章于 2024-11-13 20:34:39 发布

_AI001_

最新推荐文章于 2024-11-13 20:34:39 发布

阅读量640

点赞数 6

文章标签： pytorch 人工智能 python 计算机视觉深度学习

本文链接：https://blog.youkuaiyun.com/weixin_41159766/article/details/143510212

版权

ubuntu20.04快速安装pytorch（cpu和gpu版本）

文章目录

ubuntu20.04快速安装pytorch（cpu和gpu版本）

1. CPU版本

pip3 install torch torchvision torchaudio

2. GPU版本注意不需要额外安装cuda

2.1 检查当前驱动及支持的最大cuda版本

nvidia-smi

在这里插入图片描述

2.2 pytorch下载地址根据需要选择你的版本

https://pytorch.org/

在这里插入图片描述

2.3 下载

# 比如我下载11.8版本，注意不能超过上面的最大cuda版本
pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118

# 注意如果不进行cuda底层开发，不需要额外安装cuda（也就是nvidia-cuda-toolkit）

# 查看 显示ture则证明安装完成
>>> import torch
>>> torch.cuda.is_available()
True