gpu版本 安装pytorch
时间: 2023-11-08 21:02:11 浏览: 85
Windows+PyTorch GPU版本安装
5星 · 资源好评率100%
您好!要安装PyTorch,您需要首先确定您的GPU版本。您可以通过以下代码来查看:
```python
import torch
print(torch.version.cuda)
```
该代码会输出您当前安装的PyTorch所支持的CUDA版本。请将该版本信息提供给我,以便我可以为您提供正确的安装指南。
阅读全文