mts0629/study-pytorch

Study of PyTorch

study-pytorch

Study of PyTorch

• Pytorch
  • Docs
  • Tutorials

Environment

• NVIDIA GeForce RTX 2060 (6GB VRAM)
• Windows 10
• WSL2
  • Distro: Ubuntu 18.04.6 LTS
• PyTorch Docker image (CUDA support)
  • pytorch/pytorch:2.3.0-cuda12.1-cudnn8-runtime

Dockerfile and a runner script: docker_run.sh are in docker/.

This script uses a host user for login.

$ ./docker/docker_run.sh <IMAGE_TAG>

Also the VSCode Dev Container can be used.

Prerequisites

• NVIDIA GPU Drivers
• Docker
• NVIDIA container ToolKit
  • Installing the NVIDIA Container ToolKit
• Visual Studio Code + Dev Containers extension (for Dev Container)