-
Notifications
You must be signed in to change notification settings - Fork 810
cannot run "neon --gpu cudanet examples/convnet/i1k-alexnet-fp32.yaml" on tegra k1 #51
Comments
Since you appear to be running as root on Ubuntu, can you first make sure that I'd also suggest having a look at the items in our installation FAQ: http://neon.nervanasys.com/docs/latest/faq.html |
Hi scttl, thanks for your reply. you are right about that the nvidia-smi command is not found. ps: I can ran caffe with cudnn with no problem on jetson tk1, so I guess the CUDA installation is all right |
We made a similar change in the |
thanks scttl! editing neon/backends/init.py works, but I still can't run this sample because another erro appears: 2015-07-01 04:52:31,738 INFO:batch_norm - BatchNormalization set to train mode Is memory size a problem? tegra k1 has 2GB memory. or just something else lead to this problem? any advice for me to check out what happened? |
Try reducing your batch size down to 32 and see if the problem still Is there any particular reason you are using this system to train? You On Monday, June 29, 2015, yuehusile notifications@github.com wrote:
|
root@tegra-ubuntu:/home/hsl/neon# neon --gpu cudanet examples/convnet/i1k-alexnet-fp32.yaml
WARNING:neon.util.persist:deserializing object from: examples/convnet/i1k-alexnet-fp32.yaml
WARNING:neon.datasets.imageset:Imageset initialized with dtype <type 'numpy.float32'>
2015-06-19 12:47:19,332 WARNING:neon - setting log level to: 20
Traceback (most recent call last):
File "/usr/local/bin/neon", line 240, in
experiment, result, status = main()
File "/usr/local/bin/neon", line 202, in main
device_id=args.device_id)
File "/usr/local/lib/python2.7/dist-packages/neon/backends/init.py", line 157, in gen_backend
raise RuntimeError("Can't find CUDA capable GPU")
RuntimeError: Can't find CUDA capable GPU
I use a tegra k1 GPU
The text was updated successfully, but these errors were encountered: