1. 仮想環境を作って有効化

python3 -m venv .venv
source .venv/bin/activate

2. Pythonインタプリタを.venvにする

  1. Ctrl + Shift + P

  2. Python: Select Interpreter を選択

  3. .venv を選択

3. pipを更新、CPU版でPyTorchを入れる

python -m pip install --upgrade pip
pip install torch torchvision torchaudio

4. 動作確認

test_torch.pyを作成,

import torch

print(torch.__version__)
print("cuda available:", torch.cuda.is_available())
x = torch.tensor([1.0, 2.0, 3.0])
print(x * 2)

実行:

python3 test_torch.py
2.10.0+cu128
cuda available: True
tensor([2., 4., 6.])

などとなればOK.

5. ライブラリを追加

pip install numpy pandas matplotlib scikit-learn jupyter ipykernel

最小フォルダ構成例

dl-test/
├── .venv/
├── data/
├── notebooks/
├── src/
│   └── train.py
├── test_torch.py
└── requirements.txt