search

CrayLabs / SmartSim

SmartSim Infrastructure Library.
BSD 2-Clause "Simplified" License
219 stars 36 forks source link

dockerfiles for gpu with cudnn are missing #606

Open laszewsk opened 4 weeks ago

laszewsk commented 4 weeks ago

Description

Dockerfiles for smartsim using gpus with cudnn are missing

Justification

everyone that wants to use smartsim with gpus and docker

Implementation Strategy

the dockerfiles in https://github.com/CrayLabs/SmartSim/tree/develop/docker only contain device=cpu, but if you change it to gpu, the cudnn library is missing so smartsim does not work.

I suggest adding a new dir gpu in which a proper gpu based Dockerfile for this is placed, one coupd distingusih gpu-nvidia gpu-rocm if needed. I only need nvidia

ashao commented 3 weeks ago

Until recently, Nvidia did not have an 'official' pathway to download cuDNN without registering with their developer zone. This now seems possible, so we could explore the possibility of updating our tutorial Docker container to support this.