文件名称:keras-cntk-docker:适用于Nvidia-docker的keras + cntk的Docker容器
文件大小:1.47MB
文件格式:ZIP
更新时间:2024-05-29 10:26:33
Python
keras-cntk-docker 用于NVIDIA + docker的keras + cntk的Docker容器。 基于Durgesh Mankekar的最小+相应的。 该容器包含CNTK,TensorFlow和Keras。 当Keras在容器中运行时,默认情况下CNTK将是选定的后端。 仅建议在GPU实例上运行。 用法 要从头开始在GPU实例上设置CUDA +容器驱动程序,请运行以下命令: curl -O -s https://raw.githubusercontent.com/minimaxir/keras-cntk-docker/master/deps_nvidia_docker.sh sudo sh deps_nvidia_docker.sh 几天后,您可能需要再次运行该脚本(由于主机硬件上的GPU更新) 要在容器中运行深度学习脚本: sudo docker run
【文件预览】:
keras-cntk-docker-master
----Dockerfile(1KB)
----deps_nvidia_docker.sh(2KB)
----cntk_keras.gif(263KB)
----LICENSE(1KB)
----lstm_text_generation.py(3KB)
----README.md(3KB)
----Dockerfile-cpu-compiled(3KB)
----mnist_mlp.py(2KB)
----jupyter_notebook_config.py(1KB)
----run_jupyter.sh(733B)
----Dockerfile-cpu(1KB)
----get-pip.py(1.52MB)
----lstm_benchmark.py(2KB)