Linux系统如何下载GPU版本的PyTorch
如果您使用的是Linux系统,则可以使用以下命令下载GPU版本的PyTorch:
pip3 install torch torchvision torchaudio -f https://download.pytorch.org/whl/cu111/torch_stable.html
这将下载并安装适用于CUDA 11.1的最新版本的PyTorch,包括'torch'、'torchvision'和'torchaudio'等库。如果您使用的是其他版本的CUDA,请在URL中相应地更改'cu111'为'cuXXX'。
原文地址: http://www.cveoy.top/t/topic/lHr7 著作权归作者所有。请勿转载和采集!