pytorch源代码编译


相信很多朋友们安装pytorch时都是使用conda或者pip直接进行安装,但是如果想要了解pytorch底层的CPP实现或者修改底层C++的代码就需要拉取pytorch官方仓库进行源代码编译

所以下面我来介绍下如何从源代码编译pytorch以及相关的踩坑记录

pytorch源码编译

环境版本

cuda 10.0 pytorch1.4 python3.6

编译步骤

conda create -n name python=3.6#创建conda环境
#conda remove -n py39 --all#删除 
conda install numpy ninja pyyaml mkl mkl-include setuptools cmake cffi typing#安装依赖库
conda install -c pytorch magma-cuda100#安装对应cuda依赖
git clone --branch release/1.4  https://g
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值