PyTorch2.0.0版本是存在的,我是通过下面这条指令安装的:pip install torch==2.0.0+cu117 torchvision==0.15.1+cu117 torchaudio==2.0.1 --index-url https://download.pytorch.org/whl/cu117
时间: 2023-09-16 08:09:08 浏览: 505
torch_scatter-2.1.2+pt20cu117-cp39-cp39-win_amd64whl.zip
非常抱歉,我之前的回答有误。确实存在 PyTorch 2.0.0 版本,我对此深表歉意。
如果你已经成功安装了 PyTorch 2.0.0 + cu117,并且你的 GPU 支持 CUDA 11.7,那么你应该可以使用 optimizer 模块。你可以使用以下代码检查你的 PyTorch 和 CUDA 版本:
```python
import torch
print(torch.__version__)
print(torch.version.cuda)
```
请注意,PyTorch 2.0.0 版本是比较早期的版本,可能存在不兼容的问题。建议使用较新的版本,如 PyTorch 1.7.0 或 1.8.0 版本。
阅读全文