/pytorch-opencv-vscode

Template repository for PyTorch, OpenCV and NLOpt development in Visual Studio Code. Includes CUDA support.

Primary LanguageShellMIT LicenseMIT

pytorch-opencv-vscode

Requirements

  • CUDA capable GPU
  • nvidia-docker runtime

Features

Template repository that contains:

torch==1.8.1+cu111
torchvision==0.9.1+cu111
torchaudio==0.8.1 
opencv-python==4.5.1.48
nlopt==2.7.0
  • CUDA 11.1
  • Python 3.8 (with intellisense)
  • X11 Redirection

Usage

  • Click Use this template to create your repository.
  • Open this folder in VSCode.