zhanima
zhanima
line 19 in train.py change os.environ["CUDA_VISIBLE_DEVICES"] = "3" to os.environ["CUDA_VISIBLE_DEVICES"] = "0" if you train with gpu
> > line 19 in train.py > > change os.environ["CUDA_VISIBLE_DEVICES"] = "3" to > > os.environ["CUDA_VISIBLE_DEVICES"] = "0" if you train with gpu > > I tried ,but still >...
> **If this is your fist gpu based machine learning project, check your cuda and cudnn installation. they should be installed in pair with your gpu type number and fill...