windy_z

@yotaroshimose bro，my english is not very good ,so maybe i wrongly understand your question,the following is my advice you can add a statement -> CUDA_VISIBLE_DEVICES=0,1 (i have 4 GPU but...

ｈａｖｅ　ｔｈｅ　ｓａｍｅ　ｐｒｏｂｌｅｍ ｗａｉｔｉｎｇ　ｆｏｒ　ａ　ｇｏｏｄ　ｍａｎ　ｔｏ　ｒｅｐｌｙ

resloved! it's about CUDA_HOME you should modify your /.bashrc and add : export CUDA_HOME = your cuda path if anyone face the same problem contact me and help you

same question i meet wish publisher will slove