pytorch安装linux

最新推荐文章于 2024-08-08 19:56:43 发布

CloudCver

最新推荐文章于 2024-08-08 19:56:43 发布

阅读量236

点赞数

CC 4.0 BY-SA版权

分类专栏： Python 计算机视觉CV

本文链接：https://blog.csdn.net/CLOUD_J/article/details/104125432

Python 同时被 2 个专栏收录

24 篇文章

订阅专栏

计算机视觉CV

19 篇文章

订阅专栏

本文提供了针对不同CUDA版本和仅CPU环境下安装PyTorch及torchvision的详细指导，包括使用清华源进行快速安装的方法。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

总结装机必要包

pytorch安装linux

# CUDA 10.0
pip install torch==1.2.0 torchvision==0.4.0 -i https://pypi.tuna.tsinghua.edu.cn/simple

# CUDA 9.2
pip install torch==1.2.0+cu92 torchvision==0.4.0+cu92 -i https://pypi.tuna.tsinghua.edu.cn/simple

# CPU only
pip install torch==1.2.0+cpu torchvision==0.4.0+cpu -i https://pypi.tuna.tsinghua.edu.cn/simple