谢谢各位,我这才明白原来环境不用一模一样,直接跑就是了。然后的话只安装了一个 transformers 就可以跑了。我之前其实也跑通过,但是一分钟不到 colab 就崩溃了,所以我以为环境一定要大概相同才能跑。
然后刚才直接跑了一下,还是一分钟不到就崩溃了。
warning 信息只有
```
2023-03-14 02:02:56.805401: I tensorflow/core/platform/
cpu_feature_guard.cc:193] This TensorFlow binary is optimized with oneAPI Deep Neural Network Library (oneDNN) to use the following CPU instructions in performance-critical operations: AVX2 FMA
To enable them in other operations, rebuild TensorFlow with the appropriate compiler flags.
2023-03-14 02:03:00.821268: W tensorflow/compiler/xla/stream_executor/platform/default/
dso_loader.cc:64] Could not load dynamic library 'libnvinfer.so.7'; dlerror: libnvinfer.so.7: cannot open shared object file: No such file or directory; LD_LIBRARY_PATH: /usr/lib64-nvidia
2023-03-14 02:03:00.822404: W tensorflow/compiler/xla/stream_executor/platform/default/
dso_loader.cc:64] Could not load dynamic library 'libnvinfer_plugin.so.7'; dlerror: libnvinfer_plugin.so.7: cannot open shared object file: No such file or directory; LD_LIBRARY_PATH: /usr/lib64-nvidia
2023-03-14 02:03:00.822449: W tensorflow/compiler/tf2tensorrt/utils/
py_utils.cc:38] TF-TRT Warning: Cannot dlopen some TensorRT libraries. If you would like to use Nvidia GPU with TensorRT, please make sure the missing libraries mentioned above are installed properly.
```
然后关于显卡的话,由于之前 colab 崩溃,我也以为是显卡配置。我问过我的导师,他说 T4 足够了。