提交 c75008a3 编写于 作者: 刘托

Merge branch 'update_dockerfile' into 'master'

update dockerfile to support onnx, and fix typo

See merge request !892
...@@ -22,6 +22,7 @@ RUN apt-get install -y --no-install-recommends \ ...@@ -22,6 +22,7 @@ RUN apt-get install -y --no-install-recommends \
python-dev \ python-dev \
python-pip \ python-pip \
protobuf-compiler \ protobuf-compiler \
libprotoc-dev \
rsync \ rsync \
software-properties-common \ software-properties-common \
unzip \ unzip \
...@@ -95,7 +96,6 @@ RUN apt-get install -y --no-install-recommends \ ...@@ -95,7 +96,6 @@ RUN apt-get install -y --no-install-recommends \
android-tools-adb android-tools-adb
RUN pip install -i http://pypi.douban.com/simple/ --trusted-host pypi.douban.com setuptools RUN pip install -i http://pypi.douban.com/simple/ --trusted-host pypi.douban.com setuptools
RUN pip install -i http://pypi.douban.com/simple/ --trusted-host pypi.douban.com \ RUN pip install -i http://pypi.douban.com/simple/ --trusted-host pypi.douban.com \
tensorflow==1.8.0 \
"numpy>=1.14.0" \ "numpy>=1.14.0" \
scipy \ scipy \
jinja2 \ jinja2 \
...@@ -104,6 +104,19 @@ RUN pip install -i http://pypi.douban.com/simple/ --trusted-host pypi.douban.com ...@@ -104,6 +104,19 @@ RUN pip install -i http://pypi.douban.com/simple/ --trusted-host pypi.douban.com
pycodestyle==2.4.0 \ pycodestyle==2.4.0 \
filelock filelock
# Install Tensorflow
RUN pip install -i http://pypi.douban.com/simple/ --trusted-host pypi.douban.com \
tensorflow==1.8.0
# Install pytorch (refer to: https://pytorch.org/get-started/locally/)
RUN pip install future==0.17.1 torchvision_nightly
RUN pip install torch_nightly -f https://download.pytorch.org/whl/nightly/cpu/torch_nightly.html
# Install ONNX
RUN pip install -i http://pypi.douban.com/simple/ --trusted-host pypi.douban.com \
onnx==1.3.0 \
onnx-tf==1.2.0
# fix docker in docker error: `error while loading shared libraries: libltdl.so.7` # fix docker in docker error: `error while loading shared libraries: libltdl.so.7`
# refer to: https://github.com/jenkinsci/docker/issues/506 # refer to: https://github.com/jenkinsci/docker/issues/506
RUN apt-get install -y libltdl7 RUN apt-get install -y libltdl7
...@@ -425,7 +425,7 @@ OpenCLRuntime::OpenCLRuntime( ...@@ -425,7 +425,7 @@ OpenCLRuntime::OpenCLRuntime(
} }
device_->getInfo(CL_DEVICE_GLOBAL_MEM_CACHE_SIZE, device_->getInfo(CL_DEVICE_GLOBAL_MEM_CACHE_SIZE,
&device_gloabl_mem_cache_size_); &device_global_mem_cache_size_);
device_->getInfo(CL_DEVICE_MAX_COMPUTE_UNITS, device_->getInfo(CL_DEVICE_MAX_COMPUTE_UNITS,
&device_compute_units_); &device_compute_units_);
...@@ -464,7 +464,7 @@ cl::CommandQueue &OpenCLRuntime::command_queue() { return *command_queue_; } ...@@ -464,7 +464,7 @@ cl::CommandQueue &OpenCLRuntime::command_queue() { return *command_queue_; }
Tuner<uint32_t> *OpenCLRuntime::tuner() { return tuner_.get(); } Tuner<uint32_t> *OpenCLRuntime::tuner() { return tuner_.get(); }
uint64_t OpenCLRuntime::device_global_mem_cache_size() const { uint64_t OpenCLRuntime::device_global_mem_cache_size() const {
return device_gloabl_mem_cache_size_; return device_global_mem_cache_size_;
} }
uint32_t OpenCLRuntime::device_compute_units() const { uint32_t OpenCLRuntime::device_compute_units() const {
......
...@@ -147,7 +147,7 @@ class OpenCLRuntime { ...@@ -147,7 +147,7 @@ class OpenCLRuntime {
std::string platform_info_; std::string platform_info_;
std::string precompiled_binary_platform_info_; std::string precompiled_binary_platform_info_;
bool out_of_range_check_; bool out_of_range_check_;
uint64_t device_gloabl_mem_cache_size_; uint64_t device_global_mem_cache_size_;
uint32_t device_compute_units_; uint32_t device_compute_units_;
}; };
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册