mitmul/ssai-cnn

Updated environment versions and instructions

Opened this issue · 3 comments

Vol-i commented

Hi,
I'm trying to train my own dataset but failed to deploy after many different CUDA, CUDNN, CuPy, Python version trials on Ubuntu 16.04 on NVIDIA GTX-950M.

Is there any updated environment versions for CUDA, CUDNN, CuPy, Python, Chainer, NumPy?

Cheers,

@Vol-i
Thank you for using my code. To be honest, currently, I don't have enough time to update the versions of packages used in this repo. How about using Nvidia docker to construct the environment? This comment might be useful: #20 (comment)

Here's a Dockerfile created by @Skylion007: #5

Vol-i commented

Thanks for the code and info, I'll give it a try.
Cheers!