【pytorch基础】如何指定GPU运行程序
摘要:cmd CUDA_VISIBLE_DEVICES=1 python main.py cmd CUDA_VISIBLE_DEVICES=5, 6, 7 python main.py cmd: c++ CUDA_VISIBLE_DEVICES=1 ./maincpp 但是,在程序中指定gpu并没有正确运
阅读全文
posted @ 2022-01-18 16:02
Dive in CV/DL/ML/Algorithm.
posted @ 2022-01-18 16:02