search

mitmul / ssai-cnn

Semantic Segmentation for Aerial / Satellite Images with Convolutional Neural Networks including an unofficial implementation of Volodymyr Mnih's methods
http://www.ingentaconnect.com/content/ist/jist/2016/00000060/00000001/art00003
MIT License
260 stars 75 forks source link

Updated environment versions and instructions #22

Open Vol-i opened 6 years ago

Vol-i commented 6 years ago

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,

mitmul commented 6 years ago

@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: https://github.com/mitmul/ssai-cnn/issues/20#issuecomment-373936932

mitmul commented 6 years ago

Here's a Dockerfile created by @Skylion007: https://github.com/mitmul/ssai-cnn/pull/5

Vol-i commented 6 years ago

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