Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

cudatoolkit 10.1 #18

Open
wants to merge 6 commits into
base: master
Choose a base branch
from
Open

cudatoolkit 10.1 #18

wants to merge 6 commits into from

Conversation

jjhelmus
Copy link
Contributor

@jjhelmus jjhelmus commented May 30, 2019

Recipe and modification to the build.py script needed for cudatoolkit 10.1 update 1.
Two new libraries are included in this release, libcublasLt and libnvjpeg (linux only), both are distributable under the same license as other libraries in the previous release.
The linux run file requires that the extraction path be passed via the --installpath argument
The cublas library version is 10.2 not 10.1.

Add entries for CUDA 10.1 to build.py.  10.1 update 1 added two new
libraries, libcublasLt and libnvjpeg.

The linux run file requires the extraction path to be specified using:
--installpath=path rather than the previous --toolkitpath path.
@jjhelmus
Copy link
Contributor Author

I not yet tested this on Windows, once I have I will remove the WIP label.

@jjhelmus
Copy link
Contributor Author

Added few minor changes for the Windows build.

@jjhelmus jjhelmus changed the title WIP: cudatoolkit 10.1 cudatoolkit 10.1 May 31, 2019
@jjhelmus
Copy link
Contributor Author

This PR needs some more work as the cuda_10.1.168_418.67_linux.run file does not support CentOS 6. There is a different installer for RHEL6/CentOS 6, cuda_10.1.168_418.67_rhel6.run but is behaves differently and will need some changes.

@jjhelmus jjhelmus changed the title cudatoolkit 10.1 WIP: cudatoolkit 10.1 May 31, 2019
Use the rhel6 runfile for CUDA 10.1. The cublas libraries are removed if
a toolkit install is performed, rather extract the embedded cuda blob
and extract the files from there.
@jjhelmus jjhelmus changed the title WIP: cudatoolkit 10.1 cudatoolkit 10.1 Jun 3, 2019
@jjhelmus
Copy link
Contributor Author

jjhelmus commented Jun 3, 2019

This is working again. The linux install requires a bit of work as the standard installer does not support RHEL6/CentOS 6. The installer which does support these distributions does not extract the libcublas libraries when the --toolkit argument is provided. Instead the script extracts the internal binary blobs and extracts the needed files from one of those files.

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.

1 participant