当前位置: article > 正文

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

作者：2023面试高手 | 2024-06-03 13:03:33

踩

cuda 11.2 pytorch

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.10.1 -f https://download.pytorch.org/whl/to

声明：本文内容由网友自发贡献，不代表【wpsshop博客】立场，版权归原作者所有，本站不承担相应法律责任。如您发现有侵权的内容，请联系我们。转载请注明出处：https://www.wpsshop.cn/w/2023面试高手/article/detail/667352