-
设置使用gpu编号
#方法一 os.environ["CUDA_DEVICE_ORDER"]="PCI_BUS_ID" os.environ["CUDA_VISIBLE_DEVICES"]="1" #指定使用gpu编号 #方法二 caffe.set_device(1) #指定使用的gpu编号 caffe.set_mode_gpu()
方法一始终使用指定的gpu计算,而方法二在执行前向传播时会使用gpu0
[原创]caffe技巧
最新推荐文章于 2022-11-07 15:51:39 发布
设置使用gpu编号
#方法一
os.environ["CUDA_DEVICE_ORDER"]="PCI_BUS_ID"
os.environ["CUDA_VISIBLE_DEVICES"]="1" #指定使用gpu编号
#方法二
caffe.set_device(1) #指定使用的gpu编号
caffe.set_mode_gpu()
方法一始终使用指定的gpu计算,而方法二在执行前向传播时会使用gpu0