File tree Expand file tree Collapse file tree 1 file changed +12
-1
lines changed Expand file tree Collapse file tree 1 file changed +12
-1
lines changed Original file line number Diff line number Diff line change @@ -33,17 +33,28 @@ commands:
33
33
default : " cuda-toolkit-11-4"
34
34
steps :
35
35
- run :
36
- name : Install CUDA
36
+ name : Download CUDA repo certs
37
37
command : |
38
38
cd ~
39
39
40
40
wget https://developer.download.nvidia.com/compute/cuda/repos/<< parameters.os >>/<< parameters.platform >>/cuda-<< parameters.os >>.pin
41
41
sudo mv cuda-<< parameters.os >>.pin /etc/apt/preferences.d/cuda-repository-pin-600
42
42
43
+ - run :
44
+ name : Fetch CUDA keys
45
+ command : |
46
+
43
47
sudo apt-key adv --fetch-keys https://developer.download.nvidia.com/compute/cuda/repos/<< parameters.os >>/<< parameters.platform >>/3bf863cc.pub
48
+
49
+ - run :
50
+ name : Fetch CUDA packages
51
+ command : |
44
52
sudo add-apt-repository "deb https://developer.download.nvidia.com/compute/cuda/repos/<< parameters.os >>/<< parameters.platform >>/ /"
45
53
sudo apt-get update
46
54
55
+ - run :
56
+ name : Install CUDA
57
+ command : |
47
58
sudo apt-get install -y << parameters.cuda-pkg-name >>
48
59
49
60
create-env :
You can’t perform that action at this time.
0 commit comments