这里会显示出您选择的修订版和当前版本之间的差别。
两侧同时换到之前的修订记录 前一修订版 后一修订版 | 前一修订版 | ||
pytorch编译安装过程 [2019/12/21 11:56] admin |
pytorch编译安装过程 [2023/05/03 09:15] (当前版本) admin |
||
---|---|---|---|
行 1: | 行 1: | ||
+ | ====== wheel包下载地址 ====== | ||
+ | [[https://download.pytorch.org/whl/torch_stable.html|下载地址]] | ||
+ | |||
====== 下载源码 ====== | ====== 下载源码 ====== | ||
''git clone --recursive https://github.com/pytorch/pytorch | ''git clone --recursive https://github.com/pytorch/pytorch | ||
行 6: | 行 9: | ||
python setup.py install | python setup.py install | ||
'' | '' | ||
- | 如果需要限制CMAKE的cpu数量为2,使用```MAX_JOBS=2 python setup.py install```命令进行编译安装。\\ | + | 如果需要限制CMAKE的cpu数量为2,使用''MAX_JOBS=2 python setup.py install''命令进行编译安装。\\ |
若出现:''ptxas warning : Too big maxrregcount value specified 96, will be ignored''问题, | 若出现:''ptxas warning : Too big maxrregcount value specified 96, will be ignored''问题, |