美文网首页
服务器指定GPU跑程序

服务器指定GPU跑程序

作者: SunJi_ | 来源:发表于2020-04-18 14:59 被阅读0次

最近在跑模型的过程中,发现服务器的前几个GPU总是被占用,故只能指定后面几个空闲GPU运行。

1. 在.py中指定

import os
os.environ["CUDA_VISIBLE_DEVICES"] = "3"   #指定第三块gpu

2. 在命令行指定

CUDA_VISIBLE_DEVICES = 2, 3, 4  bash your_file.sh

3. 在环境变量中指定

#打开.bashrc,加入下面代码
export  CUDA_VISIBLE_DEVICES = 2, 3, 4

相关文章

  • 服务器指定GPU跑程序

    最近在跑模型的过程中,发现服务器的前几个GPU总是被占用,故只能指定后面几个空闲GPU运行。 1. 在.py中指定...

  • 服务器操作

    使用指定的GPU 服务器远程调用tensorboard 服务器配置ipython notebook 服务器操作 启...

  • (9)指定GPU

    PyTorch默认使用从0开始的GPU,如果GPU0正在运行程序,需要指定其他GPU。 有如下两种方法来指定需要使...

  • Tensorflow: InternalError: Blas

    用GPU跑程序时出现错误: 原因&解决方法: 有其他进程在用GPU。因为用的是PyCharm,没有跑其他程序,但是...

  • 指定GPU运行程序

    方法1 在python程序中import osos.environ["CUDA_VISIBLE_DEVICES"]...

  • LightGBM的GPU版本安装

    安装GPU版本 指定device=gpu后可以使用GPU 安装CUDA版本 指定device=cuda报错,提示需...

  • 技巧

    指定gpu

  • GPU

    指定GPU CUDA_VISIBLE_DEVICES=-1 python my.py运行程序时不使用GPUCUDA...

  • cuda 指定使用GPU

    指定使用GPU1

  • Tensorflow 指定 GPU 和修改日志等级

    指定 GPU TF 日志等级管理

网友评论

      本文标题:服务器指定GPU跑程序

      本文链接:https://www.haomeiwen.com/subject/ebmhphtx.html