Hugging Face's logo

Spaces:
Kurokabe
/
GANime
Runtime error

App Files Community
Kurokabe commited on
Commit
63a4947
·
1 Parent(s): ff33c79

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +1 -1
Dockerfile CHANGED
@@ -1,4 +1,4 @@
1
- FROM tensorflow/tensorflow:2.7.0-gpu-jupyter
2
  # Because of https://developer.nvidia.com/blog/updating-the-cuda-linux-gpg-repository-key/ and https://github.com/NVIDIA/nvidia-docker/issues/1631#issuecomment-1112828208
3
  RUN rm /etc/apt/sources.list.d/cuda.list
4
  RUN rm /etc/apt/sources.list.d/nvidia-ml.list
 
1
+ FROM tensorflow/tensorflow:2.7.0-gpu
2
  # Because of https://developer.nvidia.com/blog/updating-the-cuda-linux-gpg-repository-key/ and https://github.com/NVIDIA/nvidia-docker/issues/1631#issuecomment-1112828208
3
  RUN rm /etc/apt/sources.list.d/cuda.list
4
  RUN rm /etc/apt/sources.list.d/nvidia-ml.list