pip install torch==1.9.1+cu111 -f https://download.pytorch.org/whl/torch_stable.html和pip install torch==1.9.1有什么区别
时间: 2024-01-11 09:02:30 浏览: 235
torch==1.9.0+cu111&torchaudio==0.9.0 &torchvision==0.10.0+cu111
5星 · 资源好评率100%
`pip install torch==1.9.1` 安装的是不带 CUDA 支持的 PyTorch 版本,适用于不需要使用 GPU 的情况或者使用 CPU 运行模型的情况。
而 `pip install torch==1.9.1+cu111 -f https://download.pytorch.org/whl/torch_stable.html` 安装的是带 CUDA 11.1 支持的 PyTorch 版本,适用于在支持 CUDA 11.1 的 NVIDIA GPU 上进行加速运算。
选择哪个版本取决于您的需求和硬件配置。如果您有支持 CUDA 的 NVIDIA GPU,并且想要利用 GPU 进行加速计算,那么安装带 CUDA 支持的 PyTorch 版本是更好的选择。如果您没有支持 CUDA 的 GPU 或者不需要 GPU 加速,那么安装不带 CUDA 支持的 PyTorch 版本即可。
阅读全文