pip安装torch,CUDA为10.1版本
时间: 2024-06-10 17:05:58 浏览: 225
CUDA10.1向下兼容10.0,完美运行官方TensorFlow、Pytorch_pytorchcuda10.0,cuda10.0可以升级
5星 · 资源好评率100%
要在CUDA 10.1版本下安装torch,您可以使用以下命令:
```python
pip install torch==1.8.0+cu101 -f https://download.pytorch.org/whl/cu101/torch_stable.html
```
这将安装torch 1.8.0版本,并且支持CUDA 10.1. 感谢您的提问,是否还有其他问题需要我回答呢?
阅读全文