cuda 10.1 安装pytorch版本是多少
时间: 2023-08-10 12:55:27 浏览: 285
pytorch 查看cuda 版本方式
对于 CUDA 10.1,可以安装 PyTorch 1.5.1 版本。这个版本支持 CUDA 10.1,可以通过以下命令进行安装:
```
pip install torch==1.5.1+cu101 -f https://download.pytorch.org/whl/torch_stable.html
```
请注意,这是一个示例命令,可能需要根据您的操作系统和环境进行适当调整。
阅读全文