hollowstrawberry / kohya-colab

Accessible Google Colab notebooks for Stable Diffusion Lora training, based on the work of kohya-ss and Linaqruf
GNU General Public License v3.0
600 stars 88 forks source link

Dataset Maker Warning: Could not find TensorRT` #225

Open Tulantuna1 opened 1 day ago

Tulantuna1 commented 1 day ago

I got this error with Dataset Maker. Any idea?

Thanks `

πŸšΆβ€β™‚οΈ Launching program...

env: PYTHONPATH=/content/kohya-trainer 2024-10-04 05:07:14.082194: E external/local_xla/xla/stream_executor/cuda/cuda_fft.cc:485] Unable to register cuFFT factory: Attempting to register factory for plugin cuFFT when one has already been registered 2024-10-04 05:07:14.104023: E external/local_xla/xla/stream_executor/cuda/cuda_dnn.cc:8454] Unable to register cuDNN factory: Attempting to register factory for plugin cuDNN when one has already been registered 2024-10-04 05:07:14.110388: E external/local_xla/xla/stream_executor/cuda/cuda_blas.cc:1452] Unable to register cuBLAS factory: Attempting to register factory for plugin cuBLAS when one has already been registered 2024-10-04 05:07:14.125589: I tensorflow/core/platform/cpu_feature_guard.cc:210] This TensorFlow binary is optimized to use available CPU instructions in performance-critical operations. To enable the following instructions: AVX2 FMA, in other operations, rebuild TensorFlow with the appropriate compiler flags. 2024-10-04 05:07:15.442429: W tensorflow/compiler/tf2tensorrt/utils/py_utils.cc:38] TF-TRT Warning: Could not find TensorRT`

ShinobiiSpartan commented 1 day ago

I got the same thing, hopefully it can get fixed soon

lora1maker commented 22 hours ago

I haven't been able to use it for over a week because of this problem.