Skip to content

Add support for CUDA 11.2#559

Open
saudet wants to merge 1 commit intomasterfrom
sa_cuda112
Open

Add support for CUDA 11.2#559
saudet wants to merge 1 commit intomasterfrom
sa_cuda112

Conversation

@saudet
Copy link
Copy Markdown

@saudet saudet commented Feb 8, 2021

Fixes https://github.com/deeplearning4j/nd4j/issues/2936

I haven't tested the build, but there's no reason that it should fail given that it works with CUDA 11.1

/cc @camdenorrb @cyberbeat

Signed-off-by: Samuel Audet <samuel.audet@gmail.com>
@cyberbeat
Copy link
Copy Markdown

I would be very glad if these changes would land in a beta8 as soon as possible to get the packages from maven-central

@quickwritereader
Copy link
Copy Markdown

./change-cuda-versions.sh 11.2
mvn clean install -Dmaven.test.skip -Dlibnd4j.chip=cuda -Dlibnd4j.cuda=11.2 -Dlibnd4j.compute=auto

the build was successful for me

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

3 participants