Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
shy_201992
openpose
提交
5c5d80f9
O
openpose
项目概览
shy_201992
/
openpose
与 Fork 源项目一致
从无法访问的项目Fork
通知
7
Star
0
Fork
0
代码
文件
提交
分支
Tags
贡献者
分支图
Diff
Issue
0
列表
看板
标记
里程碑
合并请求
0
Wiki
0
Wiki
分析
仓库
DevOps
项目成员
Pages
O
openpose
项目概览
项目概览
详情
发布
仓库
仓库
文件
提交
分支
标签
贡献者
分支图
比较
Issue
0
Issue
0
列表
看板
标记
里程碑
合并请求
0
合并请求
0
Pages
分析
分析
仓库分析
DevOps
Wiki
0
Wiki
成员
成员
收起侧边栏
关闭侧边栏
动态
分支图
创建新Issue
提交
Issue看板
体验新版 GitCode,发现更多精彩内容 >>
提交
5c5d80f9
编写于
1月 13, 2021
作者:
G
Gines Hidalgo
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
Updated cudnn/cuda script style
Signed-off-by:
N
Gines Hidalgo
<
gineshidalgo99@gmail.com
>
上级
708b1b57
变更
2
显示空白变更内容
内联
并排
Showing
2 changed file
with
6 addition
and
6 deletion
+6
-6
scripts/ubuntu/install_cuda.sh
scripts/ubuntu/install_cuda.sh
+5
-5
scripts/ubuntu/install_cudnn.sh
scripts/ubuntu/install_cudnn.sh
+1
-1
未找到文件。
scripts/ubuntu/install_cuda.sh
浏览文件 @
5c5d80f9
...
...
@@ -4,20 +4,20 @@ echo "NOTE: This script assumes Ubuntu 20 or 18 (Nvidia Graphics card >= 10XX),
echo
"Otherwise, install it by yourself or it might fail."
# Install CUDA 8.0
ubuntu_version
=
"
$(
lsb_release
-r
)
"
UBUNTU_VERSION
=
"
$(
lsb_release
-r
)
"
sudo
apt-get update
&&
sudo
apt-get
install
wget
-y
--no-install-recommends
if
[[
$
ubuntu_version
==
*
"14."
*
]]
;
then
if
[[
$
UBUNTU_VERSION
==
*
"14."
*
]]
;
then
wget
-c
"https://developer.nvidia.com/compute/cuda/8.0/Prod2/local_installers/cuda-repo-ubuntu1404-8-0-local-ga2_8.0.61-1_amd64-deb"
sudo
dpkg
--install
cuda-repo-ubuntu1404-8-0-local-ga2_8.0.61-1_amd64-deb
sudo
apt-get update
sudo
apt-get
install
cuda-8-0
elif
[[
$
ubuntu_version
==
*
"16."
*
]]
;
then
elif
[[
$
UBUNTU_VERSION
==
*
"16."
*
]]
;
then
wget
-c
"https://developer.nvidia.com/compute/cuda/8.0/Prod2/local_installers/cuda-repo-ubuntu1604-8-0-local-ga2_8.0.61-1_amd64-deb"
sudo
dpkg
--install
cuda-repo-ubuntu1604-8-0-local-ga2_8.0.61-1_amd64-deb
sudo
apt-get update
sudo
apt-get
install
cuda-8-0
# Install CUDA 10.0
elif
[[
$
ubuntu_version
==
*
"18."
*
]]
;
then
elif
[[
$
UBUNTU_VERSION
==
*
"18."
*
]]
;
then
wget
-c
"https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64/cuda-ubuntu1804.pin"
sudo mv
cuda-ubuntu1804.pin /etc/apt/preferences.d/cuda-repository-pin-600
wget http://developer.download.nvidia.com/compute/cuda/10.2/Prod/local_installers/cuda-repo-ubuntu1804-10-2-local-10.2.89-440.33.01_1.0-1_amd64.deb
...
...
@@ -26,7 +26,7 @@ elif [[ $ubuntu_version == *"18."* ]]; then
sudo
apt-get update
sudo
apt-get
-y
install
cuda
# Install CUDA 11.0
elif
[[
$
ubuntu_version
==
*
"20."
*
]]
;
then
elif
[[
$
UBUNTU_VERSION
==
*
"20."
*
]]
;
then
wget
-c
"https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/cuda-ubuntu2004.pin"
sudo mv
cuda-ubuntu2004.pin /etc/apt/preferences.d/cuda-repository-pin-600
wget
"https://developer.download.nvidia.com/compute/cuda/11.0.3/local_installers/cuda-repo-ubuntu2004-11-0-local_11.0.3-450.51.06-1_amd64.deb"
...
...
scripts/ubuntu/install_cudnn.sh
浏览文件 @
5c5d80f9
...
...
@@ -3,7 +3,7 @@
echo
"This script assumes Ubuntu 16 or 14 and Nvidia Graphics card up to 10XX. Otherwise, it will fail."
# Install cuDNN 5.1
ubuntu_version
=
"
$(
lsb_release
-r
)
"
UBUNTU_VERSION
=
"
$(
lsb_release
-r
)
"
if
[[
$UBUNTU_VERSION
==
*
"14."
*
]]
||
[[
$UBUNTU_VERSION
==
*
"15."
*
]]
||
[[
$UBUNTU_VERSION
==
*
"16."
*
]]
;
then
CUDNN_URL
=
"http://developer.download.nvidia.com/compute/redist/cudnn/v5.1/cudnn-8.0-linux-x64-v5.1.tgz"
wget
-c
${
CUDNN_URL
}
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录