【Bug】os.environ[‘CUDA_VISIBLE_DEVICES‘]=‘0’指定GPU无法生效解决方案

在利用下面的语句指定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