机器学习实践
XiaoYang*
这个作者很懒,什么都没留下…
展开
-
xgboost gpu版本编译与python接口安装
1.下载源码 git clone --recursive https://github.com/dmlc/xgboost –recursive 递归clone,不加的话gputreeshape文件夹会为空,导致GPU版本编译失败 2.编译 2.1依赖工具版本 cmake version 3.16.3 GNU Make 4.2.1 2.2 编译 在xgboost目录 mkdir build cd build cmake .. -DUSE_CUDA=ON make -j4 3.安装python接口 3.1原创 2020-12-27 18:16:48 · 391 阅读 · 0 评论 -
TensorFlow2.4 问题集合
1.Train 1.1 内存分配问题 报错信息 Could not create cudnn handle: CUDNN_STATUS_INTERNAL_ERROR 解决方法 gpu_options = tf.compat.v1.GPUOptions(per_process_gpu_memory_fraction=1.0) config = tf.compat.v1.ConfigProto(gpu_options=gpu_options) config.gpu_options.allow_growth原创 2020-12-27 17:43:37 · 510 阅读 · 1 评论 -
Ubuntu20.04 DL环境配置
1.Ubuntu20.04 1.1 apt 源 deb http://mirrors.aliyun.com/ubuntu/ focal main restricted universe multiverse deb-src http://mirrors.aliyun.com/ubuntu/ focal main restricted universe multiverse deb http://mirrors.aliyun.com/ubuntu/ focal-security main restricted原创 2020-12-27 16:50:27 · 320 阅读 · 0 评论