corvus-dotnet / corvus-dotnet/Corvus.Python

Investigate perf improvements using NVidia plugin

Open
#30 2 comments 0 reactions 0 assignees View on GitHub
Dominant language
Python
Stars
0
Forks
0
PR merge metrics
No merged PRs in 30d

Description

To utilise the NVidia GPU, use:

```
"runArgs": [
"--gpus",
"all"
],
```

in the devcontainer,

and then:

```
builder = (
SparkSession.builder.appName(workload_name)
.master("local[*]")
...
.config("spark.plugins", "com.nvidia.spark.SQLPlugin")
)

extra_packages = ["com.nvidia:rapids-4-spark_2.12:24.06.0"]

spark = configure_spark_with_delta_pip(builder, extra_packages=extra_packages).getOrCreate()
```

Contributor guide

No contributing guide indexed for this repository

Assessment

This issue has not been assessed yet.

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.