pytorch编译安装过程
差别
这里会显示出您选择的修订版和当前版本之间的差别。
| 两侧同时换到之前的修订记录 前一修订版 后一修订版 | 前一修订版 | ||
|
pytorch编译安装过程 [2019/12/21 11:55] 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''问题， | ||
| 行 21: | 行 24: | ||
| 修改为： | 修改为： | ||
| - | ''... | + | ... |
| - | CUDA8_GENCODE = -gencode=arch=compute_35,code=sm_35 \ | + | CUDA8_GENCODE = -gencode=arch=compute_35,code=sm_35 \ |
| - | -gencode=arch=compute_50,code=sm_50 \ | + | -gencode=arch=compute_50,code=sm_50 \ |
| - | -gencode=arch=compute_60,code=sm_60 \ | + | -gencode=arch=compute_60,code=sm_60 \ |
| - | -gencode=arch=compute_61,code=sm_61 | + | -gencode=arch=compute_61,code=sm_61 |
| - | ...'' | + | ... |
pytorch编译安装过程.1576900555.txt.gz · 最后更改: 2020/02/01 21:40 (外部编辑)
除额外注明的地方外，本维基上的内容按下列许可协议发布： CC Attribution-Share Alike 4.0 International
