在利用下面的语句指定GPU为仅为"1",但是用nvidia-smi查看GPU使用时,仍默认为"0"
import io
import os
import time
import json
import cv2
import numpy as np
from flask import Flask, jsonify, request
from *** import **** # torch等,举个例子
os.environ['CUDA_VISIBLE_DEVICES'] = '1'
解决方案:将上述语句放到最开头:
import os
os.environ['CUDA_VISIBLE_DEVICES'] = '1'
import io
import time
import json
import cv2
import numpy as np
from flask import Flask, jsonify, request
from *** import **** # torch等,举个例子
参考:
https://blog.csdn.net/weixin_43301333/article/details/108854504