github/DeepFaceLab
1
0
Fork 0
mirror of https://github.com/iperov/DeepFaceLab.git synced 2025-07-06 21:12:07 -07:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix for colab

Browse source
This commit is contained in:
iperov 2021-04-22 21:12:42 +04:00
parent 78f12de819
commit 23130cd56a
1 changed files with 8 additions and 7 deletions
Download patch file Download diff file Expand all files Collapse all files

1
core/leras/device.py
View file

@ -92,6 +92,7 @@ class Devices(object):
@staticmethod @staticmethod
def _get_tf_devices_proc(q : multiprocessing.Queue): def _get_tf_devices_proc(q : multiprocessing.Queue):
if sys.platform[0:3] == 'win':
compute_cache_path = Path(os.environ['APPDATA']) / 'NVIDIA' / ('ComputeCache_ALL') compute_cache_path = Path(os.environ['APPDATA']) / 'NVIDIA' / ('ComputeCache_ALL')
os.environ['CUDA_CACHE_PATH'] = str(compute_cache_path) os.environ['CUDA_CACHE_PATH'] = str(compute_cache_path)
if not compute_cache_path.exists(): if not compute_cache_path.exists():