Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
chenpangpang
gpu-base-image-build
Commits
24037762
Commit
24037762
authored
Sep 19, 2024
by
chenpangpang
Browse files
feat: 更新readme
parent
e7481da7
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
6 additions
and
6 deletions
+6
-6
README.md
README.md
+6
-6
No files found.
README.md
View file @
24037762
...
...
@@ -16,8 +16,8 @@
juypterlab-pytorch:2.3.1-py3.10-cuda12.1-ubuntu22.04-devel
\
pytorch/pytorch:2.3.1-cuda12.1-cudnn8-devel
```
-
参数1:
framework
,不需要改动
-
参数2: 输出镜像名
称
-
参数1:
ide
,不需要改动
-
参数2: 输出镜像名
-
参数3: 基础镜像
-
基于
[
nvidia官方镜像
](
https://hub.docker.com/r/nvidia/cuda
)
构建镜像
```
bash
...
...
@@ -30,8 +30,8 @@
TORCHAUDIO_VERSION
=
"2.3.1"
\
CONDA_URL
=
"https://mirrors.tuna.tsinghua.edu.cn/anaconda/miniconda/Miniconda3-py38_22.11.1-1-Linux-x86_64.sh"
```
-
参数1:
framework
,不需要改动
-
参数2: 输出镜像名
称
-
参数1:
ide
,不需要改动
-
参数2: 输出镜像名
-
参数3: 基础镜像
-
TORCH_VERSION:torch版本
-
TORCHVISION_VERSION:torchvision版本
...
...
@@ -39,8 +39,8 @@
-
CONDA_URL:安装conda的url
### 相关链接
-
pytorch镜像:https://hub.docker.com/r/pytorch/pytorch/tags
-
nvidia镜像:https://hub.docker.com/r/nvidia/cuda/tags
-
pytorch镜像
(
**选择devel镜像**
)
:https://hub.docker.com/r/pytorch/pytorch/tags
-
nvidia镜像
(
**选择devel镜像**
)
:https://hub.docker.com/r/nvidia/cuda/tags
-
torch、torchvision、torchaudio、cuda版本对应:https://pytorch.org/get-started/previous-versions/
-
conda安装:https://mirrors.tuna.tsinghua.edu.cn/anaconda/miniconda/
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment