如何把pytorch2.0.0更改为pytorch1.8.1
时间: 2023-12-11 08:51:50 浏览: 141
PytorchToCaffe.rar
要将pytorch 2.0.0更改为pytorch 1.8.1,您需要卸载先前的版本,然后安装所需版本。以下是如何执行此操作的步骤:
1.卸载现有的pytorch 2.0.0版本:
```
pip uninstall torch torchvision
```
2.安装pytorch 1.8.1版本:
```
pip install torch==1.8.1 torchvision==0.9.1
```
这将安装torch 1.8.1和其相应的torchvision版本。请注意,这些版本可能需要与您的CUDA版本兼容。因此,在安装之前,请确保检查与该pytorch版本兼容的CUDA版本。
阅读全文