安装pytorch如何启用
时间: 2024-05-30 16:07:31 浏览: 127
如何正确的安装PyTorch以支持GPU步骤.txt
安装 PyTorch 的方式有多种,其中最常用的方法是使用 pip 安装,你可以使用以下命令安装:
```
pip install torch torchvision
```
安装完成后,你可以使用以下代码检查 PyTorch 是否正确安装:
```python
import torch
print(torch.__version__)
print(torch.cuda.is_available()) # 如果你有GPU的话,可以检查是否可以使用GPU加速
```
如果输出了 PyTorch 的版本号和 True(如果你有 GPU 的话),说明 PyTorch 已经成功安装。
启用 PyTorch 需要引入 torch 库,一般情况下会在代码中加入以下语句:
```python
import torch
```
在使用 PyTorch 进行开发时,你需要使用一些 PyTorch 提供的函数和类,可以通过以下方式引入:
```python
from torch import nn, optim, utils
```
阅读全文