pytorch装完怎么在jupyternotebook
时间: 2023-09-11 20:05:20 浏览: 130
在Jupyter Notebook中使用PyTorch,你需要按照以下步骤进行操作:
1. 首先,在终端中运行命令创建一个虚拟环境,并安装PyTorch和相应的依赖。可以使用以下命令:
```
conda create -n pytorch-gpu python=3.7.0
conda activate pytorch-gpu
conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch
```
2. 如果你觉得通过以上命令安装速度较慢,你可以选择离线下载PyTorch和torchvision的whl文件,然后在虚拟环境中安装。具体步骤如下:
- 访问PyTorch官方网站(https://download.pytorch.org/whl/cu113/torch_stable.html)。
- 在页面中搜索你需要的PyTorch和torchvision版本,并下载对应的whl文件。
- 将下载的whl文件传输到你的离线主机(服务器)中。
- 在已经创建好的虚拟环境中,使用pip命令安装whl文件。例如:
```
pip install torch-1.10.0 cu113-cp37-cp37m-linux_x86_64.whl
pip install torchvision-0.11.1 cu113-cp37-cp37m-linux_x86_64.whl
```
3. 在Jupyter Notebook中验证PyTorch是否安装成功。打开Jupyter Notebook,在一个代码单元格中输入以下代码并运行:
```
import torch
print(torch.__version__)
print(torch.cuda.is_available())
import torchvision
print(torchvision.__version__)
```
这样就完成了在Jupyter Notebook中使用PyTorch的安装和验证步骤。你可以在Jupyter Notebook中继续编写和运行PyTorch代码了。<span class="em">1</span><span class="em">2</span><span class="em">3</span>
#### 引用[.reference_title]
- *1* [安装pytorch成功但是在jupyter notebook中无法使用的问题](https://blog.csdn.net/qq_39033580/article/details/124249528)[target="_blank" data-report-click={"spm":"1018.2226.3001.9630","extra":{"utm_source":"vip_chatgpt_common_search_pc_result","utm_medium":"distribute.pc_search_result.none-task-cask-2~all~insert_cask~default-1-null.142^v93^chatsearchT3_1"}}] [.reference_item style="max-width: 50%"]
- *2* *3* [Jupyter Notebook与Pycharm代码连接Docker容器中的远程服务器运行](https://blog.csdn.net/qq_43966129/article/details/126843082)[target="_blank" data-report-click={"spm":"1018.2226.3001.9630","extra":{"utm_source":"vip_chatgpt_common_search_pc_result","utm_medium":"distribute.pc_search_result.none-task-cask-2~all~insert_cask~default-1-null.142^v93^chatsearchT3_1"}}] [.reference_item style="max-width: 50%"]
[ .reference_list ]
阅读全文