search

LambdaColdStorage / lambda-tensorflow-benchmark

BSD 3-Clause "New" or "Revised" License
236 stars 60 forks source link

Benchmark Issue #3

Open ghost opened 6 years ago

ghost commented 6 years ago

Really good article guys, I have the RTX 2070 here and I would like to run the test, I have installed your stack from your website, the only thing that I do after installation is updating the Nvidia drivers to recognize the RTX cards, but the tests are not working, here is the issue:

running benchmark for frameworks ['pytorch', 'tensorflow', 'caffe2'] cuda version= None cudnn version= 7201 /home/bizon/benchmark/deep-learning-benchmark-master/frameworks/pytorch/models.py:17: UserWarning: volatile was removed and now has no effect. Use with torch.no_grad(): instead. self.eval_input = torch.autograd.Variable(x, volatile=True).cuda() if precision == 'fp32' Segmentation fault

Thanks in advance

gaburipeach commented 5 years ago

I'm getting this too on my RTX 2070.