lzhengning
/
JittorMirror
mirror of https://github.com/Jittor/Jittor
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

fix: jupyter restart error

This commit is contained in:
lidongyang 2024-10-08 23:02:27 +08:00
parent 593519203b
commit 72be1396d9
1 changed files with 5 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
python/jittor_utils/install_cuda.py
View File

@ -78,8 +78,11 @@ def check_cuda_env():
with open("/proc/self/cmdline", "r") as f:
argv = f.read().split("\x00")
if len(argv[-1]) == 0: del argv[-1]
LOG.i(f"restart {sys.executable} {argv[1:]}")
os.execl(sys.executable, sys.executable, *argv[1:])
if 'ipykernel_launcher' in argv:
LOG.i(f"needed restart but not {sys.executable} {argv[1:]}, you can ignore this warning.")
else:
LOG.i(f"restart {sys.executable} {argv[1:]}")
os.execl(sys.executable, sys.executable, *argv[1:])
except:
pass