python 3.x - Can I train in tensorflow with separate CUDA version in anaconda environment

Question

I need to train a model in TensorFlow-gpu==2.3.0 which needs the CUDA version to be 10.1. But when I type 'nvidia-smi' it shows CUDA version to be 10.0.

I created a conda environment using, "conda create -n tf2-gpu tensorflow-gpu cudatoolkit=10.1" after initiating training, it throws an error as tensorflow.python.framework.errors_impl.InternalError: cudaGetDevice() failed. Status: CUDA driver version is insufficient for CUDA runtime version

How can I train using tensorflow-gpu in conda environment with another version of CUDA? And, I still need CUDA 10.0 to be there, as it helps my other training setup.

question from:https://stackoverflow.com/questions/65936263/can-i-train-in-tensorflow-with-separate-cuda-version-in-anaconda-environment

Answer 1

Waitting for answers