使用GPU環境執行VSCode Notebook

簡單的檢查CUDA連線

import torch
if torch.cuda.is.available():
    print("success")
else:
    print("failure")

如果失敗就是torch沒有使用GPU在執行

先到以下網址安裝cuda

再到以下網址使用安裝pytorch

Package選擇conda，都選好後就會給你一條指令

先在anaconda裡面建立新環境

接下來打開anaconda執行powershell prompt，啟動你的新環境

conda activate <env.name>

執行指令 Ex.

conda install pytorch torchvision torchaudio pytorch-cuda=12.4 -c pytorch -c nvidia

之後在vscode裡執行notebook右上角可以選擇環境，選擇這個環境就可以使用了