我是靠谱客的博主 丰富电灯胆,这篇文章主要介绍服务器指定GPU跑程序,现在分享给大家,希望可以做个参考。

  1. 在.py中指定
import os
os.environ["CUDA_VISIBLE_DEVICES"] = "3"   #指定第三块gpu
  1. 在命令行指定
CUDA_VISIBLE_DEVICES = 2, 3, 4  bash your_file.sh
  1. 在环境变量中指定
#打开.bashrc,#打开.bashrc,加入下面代码
export  CUDA_VISIBLE_DEVICES = 2, 3, 4

转载服务器指定GPU跑程序

最后

以上就是丰富电灯胆最近收集整理的关于服务器指定GPU跑程序的全部内容,更多相关服务器指定GPU跑程序内容请搜索靠谱客的其他文章。

本图文内容来源于网友提供,作为学习参考使用,或来自网络收集整理,版权属于原作者所有。
点赞(53)

评论列表共有 0 条评论

立即
投稿
返回
顶部