tensorflow-gpu的选择

2021-03-31  本文已影响0人  陶_306c

有两种方法:
1、CUDA_VISIBLE_DEVICES=1
2、

import os
os.environ["CUDA_VISIBLE_DEVICES"]="1"
上一篇 下一篇

猜你喜欢

热点阅读