已有cuda11.2情况下Pytorch 环境配置

最新推荐文章于 2025-02-24 17:52:10 发布

coder_yang56

最新推荐文章于 2025-02-24 17:52:10 发布

阅读量1.8w

点赞数 6

分类专栏：环境配置文章标签： python

本文链接：https://blog.csdn.net/qq_41330902/article/details/124355474

版权

本文介绍了在已有CUDA11.2的系统中如何配置PyTorch环境，提供了PyTorch官方下载链接，包括适用于Windows和Ubuntu的操作步骤。通过conda或whl文件进行安装，重点指出安装PyTorch时不必与电脑CUDA版本完全匹配，只要驱动程序兼容11.1版本即可。

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

PyTorch An open source machine learning framework that accelerates the path from research prototyping to production deployment.https://pytorch.org/

官网可用的下载链接

win/ubuntu

使用conda

# CUDA 10.2
conda install pytorch==1.9.1 torchvision==0.10.1 torchaudio==0.9.1 cudatoolkit=10.2 -c pytorch

# CUDA 11.3
conda install pytorch==1.9.1 torchvision==0.10.1 torchaudio==0.9.1 cudatoolkit=11.3 -c pytorch -c conda-forge

使用wheel

# CUDA 11.1
pip install torch==1.10.1+cu111 torchvision==0.11.2+cu111 torchaudio==0

最低0.47元/天解锁文章