pip install torch==1.7.1+cu101 torchvision==0.8.2+cu101 torchaudio==0.7.2 -f https://download.pytorch.org/whl/torch_stable.html
时间: 2025-03-17 21:12:50 浏览: 67
### 安装指定版本的 PyTorch 及其相关库
为了安装特定版本的 PyTorch (1.7.1),以及 torchvision (0.8.2) 和 torchaudio (0.7.2) 并支持 CUDA 10.1 (cu101),可以按照以下方法操作:
运行以下命令来完成安装[^1]:
```bash
pip install torch==1.7.1+cu101 torchvision==0.8.2+cu101 torchaudio==0.7.2 -f https://download.pytorch.org/whl/torch_stable.html
```
此命令会从官方 PyTorch 轮子存储页面下载并安装兼容 CUDA 10.1 的对应版本。需要注意的是,`-f` 参数指定了额外的索引 URL 来定位这些预编译好的二进制文件。
对于 Linux 用户,在实际执行前可以通过浏览器访问上述链接地址 `https://download.pytorch.org/whl/torch_stable.html` ,使用快捷键 Ctrl+F 查找关键字如 “11.0” 或者具体版本号字符串确认所需依赖项是否存在后再决定是否继续安装过程[^2]。
#### 验证安装成功与否的方法如下所示:
验证已正确安装所选组件的一个简单方式就是打开 Python 解释器或者脚本里尝试导入它们,并打印各自的版本信息出来查看。
```python
import torch
import torchvision
import torchaudio
print(f"Torch version: {torch.__version__}, TorchVision version: {torchvision.__version__}, Torchaudio version: {torchaudio.__version__}")
if torch.cuda.is_available():
print('CUDA is available')
else:
print('CUDA not detected.')
```
阅读全文
相关推荐


















