diff --git a/core/leras/device.py b/core/leras/device.py index 4d157f0..a511cab 100644 --- a/core/leras/device.py +++ b/core/leras/device.py @@ -87,7 +87,7 @@ class Devices(object): os.environ['NN_DEVICES_INITIALIZED'] = '1' os.environ['NN_DEVICES_COUNT'] = '0' - os.environ['CUDA_​CACHE_​MAXSIZE'] = '2147483647' + os.environ['CUDA_CACHE_MAXSIZE'] = '2147483647' min_cc = int(os.environ.get("TF_MIN_REQ_CAP", 35)) libnames = ('libcuda.so', 'libcuda.dylib', 'nvcuda.dll') for libname in libnames: @@ -205,4 +205,4 @@ if Devices.all_devices is None: cuda.cuCtxDetach(context) Devices.all_devices = Devices(devices) return Devices.all_devices -""" \ No newline at end of file +"""