首页 > 其他 > 详细

源码编译pytorch-1.2.0

时间:2020-05-31 13:58:12      阅读:59      评论:0      收藏:0      [点我收藏+]
ubuntu16.04自带的g++,gcc为5.4,因此另外安装4.9
sudo apt install gcc-4.9 g++-4.9 export CC="gcc-4.9"
export CXX="g++-4.9"

激活anaconda
cd /home/aistudio/miniconda3/bin
source ./activate torch_build

  
 设置cmake选项
 export CMAKE_PREFIX_PATH=/home/aistudio/miniconda3/envs/torch_build
 
 禁用nvcc
 export USE_NVCC=0
  
 编译
 cd /home/aistudio/torch_build/pytorch-1.2.0
 python setup.py clean
 python setup.py install


源码编译pytorch-1.2.0

原文:https://www.cnblogs.com/gris/p/12997084.html

(0)
(0)
   
举报
评论 一句话评论(0
关于我们 - 联系我们 - 留言反馈 - 联系我们:wmxa8@hotmail.com
© 2014 bubuko.com 版权所有
打开技术之扣,分享程序人生!