jupyter 安装pytorch
时间: 2023-10-12 11:11:46 浏览: 126
如何安装pytorch
在 Jupyter 中安装 PyTorch 可以通过以下步骤:
1. 确保已经安装了 Python 3.x 和 pip 工具。
2. 打开终端或命令提示符,输入以下命令安装 PyTorch:
```
pip install torch
```
如果需要安装 GPU 版本的 PyTorch,可以输入以下命令:
```
pip install torch torchvision torchaudio -f https://download.pytorch.org/whl/cu111/torch_stable.html
```
(其中 `cu111` 表示 CUDA 11.1 版本,如果使用其他 CUDA 版本,需要修改对应的代码)
3. 打开 Jupyter Notebook,创建一个新的 Python 3 笔记本。
4. 在代码单元格中输入以下代码,验证 PyTorch 是否已经安装成功:
```python
import torch
print(torch.__version__)
```
如果输出了 PyTorch 版本号,说明安装成功。
阅读全文