GPU Memory Usage:该进程的显存使用情况 2、os.environ来设置CUDA_VISIBLE_DEVICES os是python中非常常用的系统包,而os.environ则是设置查看系统环境变量的模块,我们可以通过这个模块把CUDA_VISIBLE_DEVICES的设置写入到环境变量中,这样在执行这个程序的时候就可以指定GPU运行了。 importos os.environ["CUDA_VISIBLE_DEVI...
sandclock Human Learning MachineLearning 新版本的CUDA跨平台查看gpu状态时,传统的"nvidia -smi"命令变成了"nvidia-smi"(没错单纯的少了空格),二者命令都无效的时候再去尝试配置环境变量的方法 发布于 2022-11-05 13:05・IP 属地天津 赞同 分享 ...