岩土识别泰迪杯代码运行问题报错

问题遇到的现象和发生背景

泰迪杯岩土识别

问题相关代码，请勿粘贴截图
# ----------------部分参数设置----------------------
# 训练数据集地址
train_img_url = 'data/rock_sample/1/'
# 将所有的图片resize成512*512
w = 512
h = 512
c = 3
learning_rate=0.0001
# GPU内存限制数
gpu_memory = 5120
# 每次执行图片数
batch_size = 3
# 网络fit迭代次数
num_epochs = 100
import os

os.environ["CUDA_DEVICES_ORDER"]="PCI_BUS_IS"

os.environ["CUDA_VISIBLE_DEVICES"] = "0"
#-------------------设置显存占用率-----------------
gpus = tf.config.experimental.list_physical_devices('GPU')

for gpu in gpus:
    tf.config.experimental.set_memory_growth(device=gpu, enable=True)
# 对需要进行限制的GPU进行设置
print(len(gpus))
tf.config.experimental.set_virtual_device_configuration(gpus[0],[tf.config.experimental.VirtualDeviceConfiguration(memory_limit=gpu_memory)])
# 查看GPU是否可用
print(tf.test.is_gpu_available())
运行结果及报错内容
IndexError                                Traceback (most recent call last)
~\AppData\Local\Temp\ipykernel_17056\2050962666.py in <module>
     25 # 对需要进行限制的GPU进行设置
     26 print(len(gpus))
---> 27 tf.config.experimental.set_virtual_device_configuration(gpus[0],[tf.config.experimental.VirtualDeviceConfiguration(memory_limit=gpu_memory)])
     28 # 查看GPU是否可用
     29 print(tf.test.is_gpu_available())

IndexError: list index out of range
我的解答思路和尝试过的方法

试了很多方法还是报错list index out of range

我想要达到的结果

可以参考我写的这篇博客哈，有详细解决办法