Google Colab运行Git项目遇ModuleNotFoundError及CUDA相关报错求助

核心问题：ModuleNotFoundError: No module named 'data' #

这是程序崩溃的直接原因，和CUDA警告无关，普通账号能运行是因为环境默认包含了模块路径。

• 原因：Python模块搜索路径没包含项目根目录/content/drive/MyDrive/apbench，导致找不到data文件夹对应的模块。
• 解决方法二选一：
  1. 在运行脚本前添加路径：

     import sys
     sys.path.append('/content/drive/MyDrive/apbench')
     

     把这段代码放在poisons_generate.py的最开头，或者在Colab单元格里先运行再启动脚本。
  2. 切换到项目根目录执行脚本：

     cd /content/drive/MyDrive/apbench
     python poisons_generate.py
     

关于TensorFlow的CUDA相关警告 #

这些警告（cuDNN/cuFFT/cuBLAS重复注册、TensorRT未找到）是Colab Pro环境的组件冲突导致的，不会让程序崩溃，普通账号没报错只是因为环境版本不同。要消除的话：

• 重新安装指定版本的TensorFlow，比如：

  pip uninstall tensorflow -y
  pip install tensorflow==2.13.0
  

• 嫌麻烦可以直接忽略，不影响核心功能。

为什么Pro账号和普通账号有差异？ #

Colab Pro和普通账号的默认环境（TensorFlow版本、CUDA工具包版本）不一样，Pro环境的模块路径默认配置没覆盖到你的项目目录，所以需要手动添加。

内容的提问来源于stack exchange，提问作者studentwithproblem