level 1
wolfzhu_2003
楼主
GPU：RTX3060 * 2
TF：2.4.1
CUDA：11.1
CUDNN：8.0.5
问题是不论设置CUDA_VISIBLE_DEVICES为1还是0，都会把任务建在GPU0上。CUDA_VISIBLE_DEVICES的设置位置已经是代码顶部了。也试过在环境中直接设置这个值，结果都一样。哪位大神知道怎么解决？多谢！
2021年08月13日 16点08分
1
TF：2.4.1
CUDA：11.1
CUDNN：8.0.5
问题是不论设置CUDA_VISIBLE_DEVICES为1还是0，都会把任务建在GPU0上。CUDA_VISIBLE_DEVICES的设置位置已经是代码顶部了。也试过在环境中直接设置这个值，结果都一样。哪位大神知道怎么解决？多谢！