pytorch-cuda-devcontainer

devcontainerを使ったPyTorch-CUDA環境

Environment

• Python Version: 3.10
• Base Image: nvidia/cuda:12.1.1-base-ubuntu22.04

事前準備

• ホストOSにNVIDIA Driver、CUDA Toolkitをインストールする。
• ホストOSにNVIDIA Container Toolkitをインストールする。

検証環境

• Ubuntu 22.04 (WSL2)
• NVIDIA GeForce RTX 3060 Ti

item	version
NVIDIA Driver	536.23
CUDA Toolkit	11.1