CODEWITHSUNDEEP
tensorflowdeep-learningpytorchenvironment-variablesvirtualenv
jabbar
jabbar

Reputation: 23

How to deal with CUDA version?

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?

Upvotes: 1

Views: 237

Answers (1)

Timbus Calin
Timbus Calin

Reputation: 15013

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

Upvotes: 1

Related Questions