Rooba
/
lightning
mirror of https://github.com/Lightning-AI/lightning.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Fix apt repo issue for docker (#3823) 

* fix docker repo issue

* docker

* docker

* docker

* no cudnn

* no cudnn

* try 16.04

Co-authored-by: Jirka Borovec <jirka@pytorchlightning.ai>
This commit is contained in:
Jeff Yang 2020-10-06 09:48:14 +06:30 committed by GitHub
parent cb2a3265e5
commit 90929fa433
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 5 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
dockers/base-cuda/Dockerfile
View File

@ -22,8 +22,11 @@
ARG CUDNN_VERSION=7
ARG CUDA_VERSION=10.1
FROM nvidia/cuda:${CUDA_VERSION}-cudnn${CUDNN_VERSION}-devel
# FROM nvidia/cuda:${CUDA_VERSION}-devel
# FROM nvidia/cuda:${CUDA_VERSION}-devel-ubuntu20.04
# FROM nvidia/cuda:${CUDA_VERSION}-cudnn${CUDNN_VERSION}-devel-ubuntu18.04
FROM nvidia/cuda:${CUDA_VERSION}-cudnn${CUDNN_VERSION}-devel-ubuntu16.04
# FROM nvidia/cuda:${CUDA_VERSION}-devel-ubuntu18.04
# FROM nvidia/cuda:${CUDA_VERSION}-devel-ubuntu16.04
ARG PYTHON_VERSION=3.7
ARG PYTORCH_VERSION=1.6