tensorflow設置gpu及gpu顯存使用

1.查看GPU      終端: nvidia-smipython 2.在終端執行程序時指定GPU    CUDA_VISIBLE_DEVICES=1   python  your_file.pysession 1:  表示用哪塊GPU.net CUDA_VISIBLE_DEVICES=1           Only device 1 will be seen CUDA_VISIBLE_DEVI
相關文章
相關標籤/搜索