Hardcoded Neural Network does not work as expected

[ccbl]

This post is deleted!

[tomekslesicki]

@Dan-Korneff I'm also getting the

 CUDNN_STATUS_NOT_SUPPORTED

error when training with CUDA. With CUDA disabled, the traning goes as expected but is very slow (as expected ;-))

Are you using CUDA to train the model? If so, how are you setting the enviorment?

Like:

conda env config vars set CUBLAS_WORKSPACE_CONFIG=:16:8
conda activate base

...or some other way?

Thank you!

[Dan Korneff]

@tomekslesicki same is happening here. Looks like there's an incompatibility with the latest CUDA driver. I'll have to tweak that

[tomekslesicki]

@Dan-Korneff the solution is to install CUDA Toolkit 11.8 and install pytorch 2.3.0 instead of the current version. Here's the install prompt:

conda install pytorch==2.3.0 torchvision==0.18.0 torchaudio==2.3.0 pytorch-cuda=11.8 -c pytorch -c nvidia