Setup Kaggle environment on extension load (#31)

* Automatically setup kaggle environment https://github.com/andreinechaev/nvcc4jupyter/issues/29

* Update package lists before installing cuda toolkit
This commit is contained in:
Cosmin Ștefan Ciocan
2024-04-03 17:23:33 +02:00
committed by GitHub
parent 4664a4ef47
commit 5741c52254
5 changed files with 79 additions and 4 deletions
+2
View File
@@ -23,6 +23,7 @@ from .parsers import (
get_parser_cuda_group_save,
)
from .path_utils import CUDA_SEARCH_PATHS, find_executable
from .setup_env import setup_environment
DEFAULT_EXEC_FNAME = "cuda_exec.out"
SHARED_GROUP_NAME = "shared"
@@ -364,5 +365,6 @@ def load_ipython_extension(shell: InteractiveShell):
"""
Method used by IPython to load the extension.
"""
setup_environment()
nvcc_plugin = NVCCPlugin(shell)
shell.register_magics(nvcc_plugin)