PyTorch于JupyterLab的環境準備是什么

這篇文章給大家介紹PyTorch于JupyterLab的環境準備是什么，內容非常詳細，感興趣的小伙伴們可以參考借鑒，希望對大家能有所幫助。

PyTorch 是目前主流的深度學習框架之一，而 JupyterLab 是基于 Web 的交互式筆記本環境。于 JupyterLab 我們可以邊記筆記的同時、邊執行 PyTorch 代碼，便于自己學習、調試或以后回顧。

安裝 Anaconda

• Anaconda: https://www.anaconda.com/products/individual#Downloads

• 北外鏡像源: https://mirrors.bfsu.edu.cn/help/anaconda/

# 激活 base 環境
conda activate base

安裝 JupyterLab

• JupyterLab: https://jupyterlab.readthedocs.io/

應該已隨 Anaconda 安裝，如下查看版本：

jupyter --version

不然，如下進行安裝：

conda install -c conda-forge jupyterlab

執行 jupyter lab 啟動，瀏覽器會打開 http://localhost:8888/ ：

版本 < 3.0 建議安裝 TOC 擴展：

jupyter labextension install @jupyterlab/toc

TOC 目錄效果如下：

創建 PyTorch 環境

• PyTorch: https://pytorch.org/

# 創建虛擬環境
conda create -n pytorch python=3.8 -y
conda activate pytorch

# 安裝 PyTorch with CUDA
#  NOTE: Python 3.9 users will need to add '-c=conda-forge' for installation
conda install pytorch==1.7.1 torchvision==0.8.2 cudatoolkit=10.2 -c pytorch -y

• torch, torchvision, python 版本兼容情況

• CUDA Toolkit 與 Nvidia 驅動版本兼容情況

• Nvidia Driver 于 Ubuntu 推薦安裝辦法

如下檢查 PyTorch 版本與 GPU 支持情況：

$ python - <<EOF
import torch, torchvision
print(torch.__version__, torch.cuda.is_available())
EOF
1.7.1 True

導入 PyTorch 環境

conda activate pytorch
# 安裝 IPython kernel for Jupyter
conda install ipykernel -y
# 導入 pytorch 虛擬環境到 ipykernel
python -m ipykernel install --user --name pytorch --display-name "Python PyTorch"

Python PyTorch 即會添加進 Launcher：

開始寫筆記

運行 JupyterLab ：

conda activate base
# 啟動 JupyterLab
jupyter lab

瀏覽器會打開 http://localhost:8888/ ，如上節圖片。

• 可于左側 File Browser 瀏覽打開 *.ipynb 筆記

• 可于右側 Launcher 創建編輯 Python PyTorch 筆記

筆記可編輯 Code 并執行：

或者 VS Code 安裝 Jupyter 擴展進行瀏覽與編輯。

注意：筆記右上角選擇的 Kernel 應為 Python PyTorch。

關于PyTorch于JupyterLab的環境準備是什么就分享到這里了，希望以上內容可以對大家有一定的幫助，可以學到更多知識。如果覺得文章不錯，可以把它分享出去讓更多的人看到。