neibla/PyTorch-Universal-Docker-Template

Template repository to build PyTorch projects from source on any version of PyTorch/CUDA/cuDNN.

Dockerfile