电脑上有多少个,程序里面指定编号到底怎么确定?
nvidia -smi
可以查看信息
如果要实时查看则用 watch -n x nvidia-smi (x为时间周期)
另外,查看cpu信息 lscpu
import os
os.environ["CUDA_VISIBLE_DEVICES"] = "2" #指定编号
(BKY-博客园)
电脑上有多少个,程序里面指定编号到底怎么确定?
nvidia -smi
可以查看信息
如果要实时查看则用 watch -n x nvidia-smi (x为时间周期)
另外,查看cpu信息 lscpu
import os
os.environ["CUDA_VISIBLE_DEVICES"] = "2" #指定编号
(BKY-博客园)