安装torch1.7.0,torchvision0.8.1
时间: 2023-11-20 10:45:38 浏览: 158
jetson nano torch1.7.0 torchvision 0.8.0
可以使用以下代码安装:
```
pip install torch==1.7.0 torchvision==0.8.1 -f https://download.pytorch.org/whl/cu110/torch_stable.html
```
此代码将安装适用于 CUDA 11.0 的 PyTorch 版本 1.7.0 和 torchvision 版本 0.8.1。如果您的 CUDA 版本不同,请将“cu110”替换为您所需的版本,例如“cu102”或“cpu”。
阅读全文