search

tianweiy / CenterPoint

MIT License
1.89k stars 458 forks source link

Nvidia Docker #17

Open iamsiddhantsahu opened 4 years ago

iamsiddhantsahu commented 4 years ago

Is possible to get the Nvidia Docker image for this project, we need not have to downgrade our Nvidia drivers, CUDA version from 11 to 10 and cuDNN and would be sure that this works. Please find below the link for Nvidia Docker https://github.com/NVIDIA/nvidia-docker

tianweiy commented 4 years ago

I never use docker in the past and I am busy with another project recently. You may be able to find some help here

tianweiy commented 4 years ago

Also, in my memory, Cuda 11 with PyTorch 1.3 seems to work with VoxelNet (Pillar gets some bug).

tianweiy commented 4 years ago

I think it should work with a bunch of cuda / cudnn / pytorch version so long as you can install spconv. Reopen if you get any issues

darrenjkt commented 3 years ago

I made a docker image for this repo - feel free to use it. https://github.com/darrenjkt/CenterPoint

tianweiy commented 3 years ago

I appreciate the help! I also want to use docker recently

tianweiy commented 3 years ago

Another dockerfile is available here https://github.com/tianweiy/CenterPoint/blob/new_release/Dockerfile which I use for the waymo challenge. This one assumes that you have spconv code in the same directory as the CenterPoint folder.