cuda + miniconda + mlflow + ML packages required for chest x-ray classification including TensorFlow
10K+
entry_points: main: parameters: epoch: {type: int, default: 3} bsize: {type: int, default: 30} max_sample: {type: int, default: 1000} command: python main.py --epoch {epoch} --bsize {bsize} --max_sample {max_sample}
conda env create -f requirements.yaml -n env_name
docker pull artinmajdi/miniconda-cuda-tensorflow:latest
conda install -c anaconda keras tensorflow-gpu
conda install -c anaconda numpy pandas matplotlib
conda install -c anaconda scikit-learn scikit-image
conda install -c anaconda psycopg2 git
pip install mlflow==1.12.1
pip install pysftp
Content type
Image
Digest
Size
5.5 GB
Last updated
over 5 years ago
docker pull artinmajdi/miniconda-cuda-tensorflow