Closed zdx3578 closed 1 year ago
ubuntu 1804 cuda driver: 470 1 pip install requirements.txt gpu error, follow: https://github.com/danijar/dreamerv3/issues/5 update Jax and https://github.com/google/jax#installation
2 pip install --upgrade protobuf==3.20.1
3 pip uninstall jax jaxlib 4 pip install --upgrade "jax[cuda11_pip]" -f https://storage.googleapis.com/jax-releases/jax_cuda_releases.html
again erro: UnfilteredStackTrace: jaxlib.xla_extension.XlaRuntimeError: INTERNAL: nvlink exited with non-zero error code 256,output: nvlink fatal : Input file '/tmp/tempfile-llpc-cab04a7-18966-5fd3b5872b691.cubin' newer than toolkit (118 vs114)
update cuda 11 ; online update cuda failed; 5 wget https://developer.download.nvidia.com/compute/cuda/11.8.0/local_installers/cuda_11.8.0_520.61.05_linux.run 6 sudo /etc/init.d/gdm3 stop 7 sudo sh cuda_11.8.0_520.61.05_linux.run
8 python example.py run train ok；
ubuntu 1804 cuda driver: 470 1 pip install requirements.txt gpu error, follow: https://github.com/danijar/dreamerv3/issues/5 update Jax and https://github.com/google/jax#installation
2 pip install --upgrade protobuf==3.20.1
3 pip uninstall jax jaxlib 4 pip install --upgrade "jax[cuda11_pip]" -f https://storage.googleapis.com/jax-releases/jax_cuda_releases.html
again erro: UnfilteredStackTrace: jaxlib.xla_extension.XlaRuntimeError: INTERNAL: nvlink exited with non-zero error code 256,output: nvlink fatal : Input file '/tmp/tempfile-llpc-cab04a7-18966-5fd3b5872b691.cubin' newer than toolkit (118 vs114)
update cuda 11 ; online update cuda failed; 5 wget https://developer.download.nvidia.com/compute/cuda/11.8.0/local_installers/cuda_11.8.0_520.61.05_linux.run 6 sudo /etc/init.d/gdm3 stop
7 sudo sh cuda_11.8.0_520.61.05_linux.run
8 python example.py run train ok；