【工具】PyTorch安装（匹配CUDA版本）

暖焱

已于 2023-06-14 09:44:53 修改

阅读量1.3k

点赞数

文章标签： pytorch 深度学习 python

于 2023-06-06 09:08:35 首次发布

本文链接：https://blog.csdn.net/guai7guai11/article/details/131060347

版权

CUDA Toolkit Archive（含PyTorch所有版本）

根据自己需要的CUDA版本和PyTorch版本找到对应的下载命令，这里以PyTorch1.10.1 CUDA 11.3 为例。

在这里插入图片描述

Linux and Windows

# CUDA 10.2
conda install pytorch==1.10.1 torchvision==0.11.2 torchaudio==0.10.1 cudatoolkit=10.2 -c pytorch

# CUDA 11.3
conda install pytorch==1.10.1 torchvision==0.11.2 torchaudio==0.10.1 cudatoolkit=11.3 -c pytorch -c conda-forge

# CPU Only
conda install pytorch==1.10.1 torchvision==0.11.2 torchaudio==0.10.1 cpuonly -c pytorch