1. 仮想環境を作って有効化
python3 -m venv .venv
source .venv/bin/activate
2. Pythonインタプリタを.venvにする
Ctrl + Shift + P
Python: Select Interpreterを選択
.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