如何在Jupyter Notebook中导入PyTorch?
时间: 2024-08-29 08:00:45 浏览: 70
genbmm:在PyTorch中用于通用矩阵乘法的CUDA内核
在Jupyter Notebook中导入PyTorch,你需要首先确保已经在当前会话中激活了包含PyTorch的Conda环境。以下是步骤:
1. 打开Jupyter Notebook,如果尚未激活特定的虚拟环境(如之前安装了PyTorch的环境),可以在Notebook的顶部菜单栏中点击 "Kernel" -> "Change kernel",然后选择正确的环境(比如`torch_env`)。
2. 确认你在活跃的环境中,输入以下Python代码行来导入PyTorch:
```python
import torch
```
这将加载整个PyTorch库及其相关的模块,包括张量处理、神经网络等功能。
如果你想导入特定模块,如`torchvision`,用于图像处理,只需要添加:
```python
import torchvision
```
如果你在导入后想查看版本信息,可以执行:
```python
print(torch.__version__)
```
这将会显示PyTorch的版本号。
阅读全文