Closed wuhaixu2016 closed 2 years ago
Hi, I have solved this problem with the following settings:
Hello, How were you able to install cuda 10.1 with tensorflow 2.7.0 as it needs cuda 11 version
@wuhaixu2016 Hello! Sorry to bother you. Have you ever encountered the following problem? I think it may be related to the version of flax.
Traceback (most recent call last):
File "lra_benchmarks/listops/train.py", line 28, in <module>
from flax.deprecated import nn
ModuleNotFoundError: No module named 'flax.deprecated'
@wuhaixu2016 This is my setting:
Hi, Thanks for your code.
I try to run this code on GPUs. But I do suffer a lot in the environment setting.
Can you provide the version of CUDA, jaxlib, jax, flax, when using GPUs for training?