cuda10.1对应的torch表
时间: 2024-05-20 19:11:34 浏览: 141
以下是 CUDA 10.1 对应的 PyTorch 版本:
| CUDA 版本 | PyTorch 版本 |
|:--------:|:-----------:|
| 10.1 | 1.4.0 |
| 10.1 | 1.5.0 |
| 10.1 | 1.6.0 |
| 10.1 | 1.7.0 |
| 10.1 | 1.8.0 |
| 10.1 | 1.9.0 |
相关问题
cuda10.1对应的torch版本
如果你使用的是 NVIDIA 的显卡,建议使用 CUDA 10.1 对应的 PyTorch 版本是 1.4.0 或更高版本。你可以使用以下命令来安装:
```
pip install torch torchvision -f https://download.pytorch.org/whl/cu101/torch_stable.html
```
其中 `-f` 参数用于指定 PyTorch 的下载源为 CUDA 10.1 版本。
cuda10.1对应的torch版本号
对应的 PyTorch 版本号应该是 1.4.0,可以通过以下命令安装:
```shell
pip install torch torchvision -f https://download.pytorch.org/whl/cu101/torch_stable.html
```
其中,`cu101` 表示 CUDA 10.1 版本。如果你的 CUDA 版本不同,需要相应地修改后面的文件路径。
阅读全文