naturomics/CapsNet-Tensorflow

dear author, can you tell me what is your device ,what is you used time with the default config;

reaIws opened this issue · 11 comments

作者您好,我想纵向比较下显卡性能,您能告诉我,在默认配置下,您得训练所用时间是多少么,我现在正在用tesla显卡训练,如果训练结束,我会告诉您,我得用时:》

                      The following table is the approximate training time
num x GPU device other configurations speed(min/epoch)
1 x 1080ti default 3
1 x K80 default 11

I also ran on a 750ti GPU, but that cards is gone and the record is lost, remember it's 8-10 min/epoch

谢谢,我是两个显卡不知道会用到几个,但是速度是 6.5min/epoch :>

You can check it with the command nvidia-smi to see how many GPU the program is using under Linux. For windows user, find it out in Task Manager(任务管理器). But by default it should use one GPU.

BTW, which tesla GPU and CPU u'r using?

ma computer has two tesla k80
cpu is E5-2630v4
it computer capability is 3.7
1080Ti's capability is 6.1
then:
when i run pyhon main.py
i get:
createing tensorflow device {/device:GPU:0}
createing tensorflow device {/device:GPU:1}

when i run nvidia-smi
it show:

GPU PID Type GPU memory usage volatile GPU-Util Temp
0 4853 C 9367MiB 83% 59C
1 4853 C 62MiB 0% 35C

So this code will only work on GPU and not on CPU?

@jackalwoods Looks like it's your tf env problem. please make sure your tf is installed correctly or a right version (the code was developed with tf 1.3, a newer one but older than 2.0 should work).