TrojanNet
TrojanNet copied to clipboard
libcublas error
I installed tensorflow-gpu==1.12.0, Keras==2.2.4, and numpy==1.17.4 when i try to run the train command, I get a libcublas error as follows: ImportError: libcublas.so.9.0: cannot open shared object file: No such file or directory
python = 3.6.9 cuda = 10.2.89 gpu = Nvidia Quadro P1000
Is there a specific CUDA or cuDNN version that I need to install to successfully run this project?