[SOLVED] How to deal with CUDA version?

Issue

How to set up different versions of CUDA in one OS?

Here is my problem: Lastest Tensorflow with GPU support requires CUDA 11.2, whereas Pytorch works with 11.3. So what is the solution to install both libraries in Windows and Ubuntu?

Solution

One solution is to use Docker Container Environment, which would only need the Nvidia Driver to be of version XYZ.AB; in this way, you can use both PyTorch and TensorFlow versions.

A very good starting point for your problem would be this one(ML-WORKSPACE) : https://github.com/ml-tooling/ml-workspace

Answered By - Timbus Calin

Answer Checked By - Candace Johnson (WPSolving Volunteer)