amd64 with gpus environement
cp .devcontainer/samples/devcontainer.amd64_gpu.json .devcontainer/devcontainer.json
cp .devcontainer/samples/Dockerfile.amd_gpu .devcontainer/Dockerfile
arm64 or non-gpu environment (support multi architectures)
cp .devcontainer/samples/devcontainer.multi.json .devcontainer/devcontainer.json
cp .devcontainer/samples/Dockerfile.multi .devcontainer/Dockerfile
Base Image
pytorch/pytorch:2.1.2-cuda12.1-cudnn8-devel
Installed Packages
- jupyter
- matplotlib
- pandas
- bs4 (beautifulsoup)
- selenium