Docker image for a base image containing machine learning tools
- Dockerhub: https://hub.docker.com/r/duckietown/dt-machine-learning-base-environment/
- Docker:
docker pull duckietown/dt-machine-learning-base-environment:daffy-amd64 - Dockerfile:
From duckietown/dt-machine-learning-base-environment:daffy-amd64
- numpy==1.18.0
- opencv-python==4.4.0.44
- tensorflow==2.2.1
- cupy-cuda101
- torch==1.7.0+cu101
- torchvision==0.8.1+cu101
- torchaudio==0.7.0
- aido-protocols-daffy
- duckietown-world-daffy
- duckietown-challenges-daffy
- cuda-cudart-dev-10-1
- cuda-compat-10-1
- cuda-libraries-10-1
- cuda-npp-10-1
- cuda-nvtx-10-1
- cuda-nvml-dev-10-1
- cuda-command-line-tools-10-1
- cuda-nvprof-10-1
- cuda-npp-dev-10-1
- cuda-libraries-dev-10-1
- cuda-minimal-build-10-1
- libnccl-dev=2.7.8-1+cuda10.1
- libnccl2=2.7.8-1+cuda10.1
- libcublas-dev=10.2.1.243-1
- libcublas10=10.2.1.243-1
- libcudnn7=7.6.5.32-1+cuda10.1
- libcudnn7-dev=7.6.5.32-1+cuda10.1
- libgl1-mesa-glx
- python3-tk