Open maxrohleder opened 3 years ago
I managed to get it to work by manually downgrading to tensorflow 2.3
:
pip install tensorflow==2.3
Maybe this could be integrated into the dependency management of pip?
The pyronn package need to be build against a specific Tensorflow version. The current packages in the pip repository are built against TF 2.3 with CUDA 10 . Currently, there are some problems with the bazel configuration preventing a version for Tensorflow 2.4 with Cuda 11 .
I agree, that this should be inetragted in the dependency management, or at least the supported and working version combinations should be statet in this repository for clarification.
@mareikethies While we are at it: Maybe the windows-binary can be made available via pip as well?
I made my build available here: https://github.com/maxrohleder/win-pyronn
Hi, I encountered the same problem... My GPU on the server is 3090, which strictly requires CUDA11. Is there any way to fix this CUDA version problem on Linux platform? @csyben @maxrohleder Many thanks!
@cocoakang You would have to recompile the layers with your desired cuda version. In theory that's not a problem if you feel comfortable with the tf build pipeline.
The pyronn package need to be build against a specific Tensorflow version. The current packages in the pip repository are built against TF 2.3 with CUDA 10 . Currently, there are some problems with the bazel configuration preventing a version for Tensorflow 2.4 with Cuda 11 .
I agree, that this should be inetragted in the dependency management, or at least the supported and working version combinations should be statet in this repository for clarification.
I've built a CUDA11 version against TF 2.5, it currently supports only py3.6 (I'm still working on newer versions), but the patches needed by CUDA 11 should be almost the same.
When installing pyronn like so:
On my system:
cuda 10.1
andcudnn8.0.4
I tested the installation:
I get the following log/error:
First I had
cuda 11.1
installed, but then I got this error:, which made me assume, that cuda 10.1 is needed by the pyronn-layers.
I suspect, that the layers were built against a specific tf version. This thread suggests that the error code implies this. Which version were the layers build against? (Maybe I am missing something) Could this be taken care of by the setup wheel?