Skip to content

Commit 646f857

Browse files
Randall Huntbrianjo
authored andcommitted
Update CUDA version to 10.2 (pytorch#1373)
* use more modern python for devcontainer * update to cuda10.2 for circleci builds Co-authored-by: Brian Johnson <[email protected]>
1 parent 5c14398 commit 646f857

File tree

4 files changed

+22
-25
lines changed

4 files changed

+22
-25
lines changed

.circleci/config.yml

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -106,7 +106,7 @@ pytorch_tutorial_build_defaults: &pytorch_tutorial_build_defaults
106106
sudo pip -q install awscli==1.16.35
107107
108108
if [ -n "${CUDA_VERSION}" ]; then
109-
DRIVER_FN="NVIDIA-Linux-x86_64-440.59.run"
109+
DRIVER_FN="NVIDIA-Linux-x86_64-440.59.run"
110110
wget "https://s3.amazonaws.com/ossci-linux/nvidia_driver/$DRIVER_FN"
111111
sudo /bin/bash "$DRIVER_FN" -s --no-drm || (sudo cat /var/log/nvidia-installer.log && false)
112112
nvidia-smi
@@ -642,4 +642,3 @@ workflows:
642642
branches:
643643
only:
644644
- master
645-

.devcontainer/Dockerfile

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,8 @@
1-
FROM python:3.6-slim
1+
FROM mcr.microsoft.com/vscode/devcontainers/python:3.8
22

3-
COPY requirements.txt requirements.txt
3+
COPY requirements.txt /tmp/pip-tmp/
44

55
RUN apt-get update && export DEBIAN_FRONTEND=noninteractive \
66
&& apt-get install git gcc unzip make -y \
7-
&& pip install --no-cache-dir -r requirements.txt
7+
&& pip3 install --disable-pip-version-check --no-cache-dir -r /tmp/pip-tmp/requirements.txt \
8+
&& rm -rf /tmp/pip-tmp

.devcontainer/devcontainer.json

Lines changed: 17 additions & 20 deletions
Original file line numberDiff line numberDiff line change
@@ -1,21 +1,18 @@
11
{
2-
"name": "PyTorch",
3-
"build": {
4-
"context": "..",
5-
"dockerfile": "Dockerfile",
6-
"args": { }
7-
},
8-
"settings": {
9-
"terminal.integrated.shell.linux": "/bin/bash",
10-
"workbench.startupEditor": "none",
11-
"files.autoSave": "afterDelay",
12-
"python.dataScience.enabled": true,
13-
"python.dataScience.alwaysTrustNotebooks": true,
14-
"python.insidersChannel": "weekly",
15-
"python.showStartPage": false
16-
},
17-
"extensions": [
18-
"ms-python.python",
19-
"lextudio.restructuredtext"
20-
]
21-
}
2+
"name": "PyTorch Tutorials",
3+
"build": {
4+
"context": "..",
5+
"dockerfile": "Dockerfile",
6+
"args": {}
7+
},
8+
"settings": {
9+
"terminal.integrated.shell.linux": "/bin/bash",
10+
"workbench.startupEditor": "none",
11+
"files.autoSave": "afterDelay",
12+
"python.dataScience.enabled": true,
13+
"python.dataScience.alwaysTrustNotebooks": true,
14+
"python.insidersChannel": "weekly",
15+
"python.showStartPage": false
16+
},
17+
"extensions": ["ms-python.python", "lextudio.restructuredtext"]
18+
}

build.sh

100644100755
File mode changed.

0 commit comments

Comments
 (0)