Add Nvidia GPU TEE Support #14
Open
Conversation
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
- Introduced a new setup_gpu function to configure GPU CC for the host system. - Added command to download GPU admin tools from the NVIDIA repository during initialization. - Updated main function to include the new setup_gpu command.
- Updated setup_gpu function to include GPU passthrough configuration. - Added a new script, gpu_passthrough.sh, to manage NVIDIA GPU passthrough for virtual machines. - Implemented functions for detecting GPUs, enabling VFIO-PCI, and configuring passthrough.
- Created nvidia-lkca.conf to manage NVIDIA module loading with ecdsa_generic and ecdh. - Ensured proper permissions for the configuration file.
- Updated VCPU count limit from 255 to 32 in launch.sh. - Modified UEFI drive options for improved compatibility. - Added GPU passthrough parameters to support NVIDIA GPUs in launch.sh. - Adjusted memory configuration in config.py from 204800 to 32768. - Enhanced base_setup.sh to install NVIDIA drivers and CUDA toolkit. - Increased tmpfs sizes for /var and /tmp in init.sh for better resource management.
PeterFarber
reviewed
Jul 1, 2025
launch.sh
Outdated
| @@ -386,7 +393,7 @@ else | |||
| fi | |||
|
|
|||
| # add number of VCPUs | |||
| [ -n "${SMP}" ] && add_opts "-smp ${SMP},maxcpus=255" | |||
| [ -n "${SMP}" ] && add_opts "-smp ${SMP},maxcpus=32" | |||
There was a problem hiding this comment.
Why is this nessasary?
PeterFarber
reviewed
Jul 1, 2025
run
Outdated
| @@ -61,13 +61,15 @@ def init(): | |||
|
|
|||
| # Download and extract SNP release tarball. | |||
| tarball = os.path.join(config.dir.build, "snp-release.tar.gz") | |||
| url = "https://github.com/SNPGuard/snp-guard/releases/download/v0.1.2/snp-release.tar.gz" | |||
| url = "https://github.com/apuslabs/snp-guard/releases/download/v6.9.0/snp-release.tar.gz" | |||
There was a problem hiding this comment.
Will upload release to ardrive and fork this repo to permaweb.
Then use the arweave tx to download
PeterFarber
reviewed
Jul 1, 2025
run
Outdated
| with requests.get(url, stream=True) as r: | ||
| r.raise_for_status() | ||
| with open(tarball, "wb") as f: | ||
| shutil.copyfileobj(r.raw, f) | ||
| run_command(f"tar -xf {tarball} -C {config.dir.build}") | ||
| run_command(f"rm {tarball}") | ||
| # Download the GPU admin tools | ||
| run_command(f"cd {config.dir.build} && git clone https://github.com/nvidia/gpu-admin-tools && cd .. ") |
There was a problem hiding this comment.
I will work the gpu-admin-tools to permaweb.
Then replace this with the forked repo.
PeterFarber
reviewed
Jul 1, 2025
scripts/base_setup.sh
Outdated
| export NEEDRESTART_MODE=a | ||
|
|
||
| echo "Installing NVIDIA CUDA repository..." | ||
| wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2204/x86_64/cuda-keyring_1.1-1_all.deb |
There was a problem hiding this comment.
Download this and upload to ardrive and use an arweave tx id to download.
- Update HyperBEAM branch from main to edge - Use dynamic SMP value for QEMU maxcpus instead of hardcoded 32 - Replace GitHub/NVIDIA URLs with Arweave for SNP release and CUDA keyring - Switch gpu-admin-tools repository from nvidia to permaweb - Add GPU setup integration to base image build process
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Add this suggestion to a batch that can be applied as a single commit.
This suggestion is invalid because no changes were made to the code.
Suggestions cannot be applied while the pull request is closed.
Suggestions cannot be applied while viewing a subset of changes.
Only one suggestion per line can be applied in a batch.
Add this suggestion to a batch that can be applied as a single commit.
Applying suggestions on deleted lines is not supported.
You must change the existing code in this line in order to create a valid suggestion.
Outdated suggestions cannot be applied.
This suggestion has been applied or marked resolved.
Suggestions cannot be applied from pending reviews.
Suggestions cannot be applied on multi-line comments.
Suggestions cannot be applied while the pull request is queued to merge.
Suggestion cannot be applied right now. Please check back later.
No description provided.