如何在jupyter notebook中使用pytorch?
时间: 2023-11-25 18:07:40 浏览: 90
基于 jupyter notebook pytorch使用卷积网络的图像10分类附完整代码+数据集可直接运行 毕业设计
5星 · 资源好评率100%
要在jupyter notebook中使用pytorch,需要先在anaconda环境中安装pytorch。可以通过以下步骤来实现:
1. 进入anaconda prompt,输入以下命令来创建一个新的conda环境:conda create --name pytorch_env
2. 激活新环境:conda activate pytorch_env
3. 安装pytorch:conda install pytorch torchvision torchaudio cpuonly -c pytorch
4. 安装jupyter notebook:conda install jupyter notebook
5. 在jupyter notebook中打开一个新的notebook,选择python3 kernel。
6. 在notebook中导入pytorch并测试代码,例如:import torch x = torch.empty(5,3) print(x)
阅读全文