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.