Tensorflow
江户川柯西い
这个作者很懒,什么都没留下…
展开
-
查看服务器的cuda版本
cat /usr/local/cuda/version.txt结果如下:原创 2020-10-18 09:52:15 · 3985 阅读 · 0 评论 -
Tensorflow中GPU和CPU的切换配置
使用CPUimport osos.environ["CUDA_DEVICE_ORDER"] = "PCI_BUS_ID"os.environ["CUDA_VISIBLE_DEVICES"] = "-1" # 这一行注释掉就是使用cpu,不注释就是使用gpu一般情况安装过GPU的会默认使用gpu使用GPUimport osos.environ["CUDA_DEVICE_ORDER"]...原创 2019-12-05 19:23:53 · 1409 阅读 · 0 评论