Bharath Kumar Adinarayan Barath19

Install the latest/stable version of the nvidia driver. For eg, 12.2 in my case

+---------------------------------------------------------------------------------------+
| NVIDIA-SMI 535.161.08             Driver Version: 535.161.08   CUDA Version: 12.2     |
|-----------------------------------------+----------------------+----------------------+
| GPU  Name                 Persistence-M | Bus-Id        Disp.A | Volatile Uncorr. ECC |
| Fan  Temp   Perf          Pwr:Usage/Cap |         Memory-Usage | GPU-Util  Compute M. |
|                                         |                      |               MIG M. |
|=========================================+======================+======================|

	#!/usr/bin/env python

	import sys, io, datetime
	from pathlib import Path
	import argparse
	from subprocess import DEVNULL, Popen, PIPE

	import rospy
	import rosbag
	from importlib import import_module



	import gym
	import tensorflow as tf
	import os
	import datetime
	import stable_baselines
	from stable_baselines.common.policies import MlpPolicy
	from stable_baselines.bench import Monitor
	from stable_baselines.common.vec_env.dummy_vec_env import DummyVecEnv