jupyter中安装pytorch
时间: 2023-10-28 19:00:04 浏览: 142
如何安装pytorch
在Jupyter Notebook中安装Pytorch的步骤如下:
1. 打开Anaconda Prompt,输入以下命令安装插件:
```
conda install nb_conda
```
根据提示输入`y`进行安装。
2. 进入创建的Pytorch环境,输入以下命令安装ipykernel:
```
conda install ipykernel
```
根据提示输入`y`进行安装。
3. 安装完成后打开Jupyter Notebook,在语言选项中会看到新增了两个选项。
4. 打开需要使用Pytorch的代码文件,点击"Kernel" -> "Change Kernel" -> 选择创建好的Pytorch环境。
5. 再次运行`import torch`,应该不会报错了。
简单总结安装步骤如下:
1. 前提:安装好Anaconda和Python3.7。
2. 创建Pytorch环境。
3. 安装Pytorch包。
4. 验证安装成功。
5. 在Jupyter Notebook中配置Pytorch环境。
详细步骤如下:
1. 打开Anaconda Prompt,输入以下命令创建Pytorch环境:
```
conda create -n Pytorch python=3.7
```
(其中Pytorch为自定义环境名称)
2. 输入以下命令激活Pytorch环境:
```
conda activate Pytorch
```
3. 根据你的CUDA版本,去Pytorch官网下载对应版本的Pytorch,并将下载命令粘贴到Anaconda Prompt中进行安装。
4. 验证是否安装成功,输入以下命令:
```
python
import torch
torch.cuda.is_available()
```
如果结果是True,则表示安装成功。
5. 在Jupyter Notebook中配置Pytorch环境,依次执行以下命令:
```
conda install nb_conda (在Anaconda Prompt的(base)环境下执行)
conda install ipykernel (在(Pytorch)环境下执行)
python -m ipykernel install --user --name pytorch --display-name "Pytorch" (在(Pytorch)环境下执行)
```
6. 打开Jupyter Notebook,创建一个Pytorch的文件,如果显示空心圆,则配置成功;如果显示实心圆,并弹出连接失败的错误提示,则需要在(Pytorch)环境下执行以下命令:
```
pip uninstall pyzmq
pip install pyzmq==19.0.2
```
完成以上步骤后,你就可以在Jupyter Notebook中使用Pytorch了。
阅读全文