pytorch编译安装过程

差别

这里会显示出您选择的修订版和当前版本之间的差别。

到此差别页面的链接

--- pytorch编译安装过程 [2019/06/18 04:31]
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
+''
-```
 ====== 编译 ======
-```
+''
 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''问题，
 将/third_party/nccl/nccl/makefiles/common.mk第28行位置处sm_30的选项去掉：
-```
-...
+  ...
-CUDA8_GENCODE = -gencode=arch=compute_30,code=sm_30 \
+  CUDA8_GENCODE = -gencode=arch=compute_30,code=sm_30 \
-                -gencode=arch=compute_35,code=sm_35 \
+                  -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
-...
+  ...
 修改为：
-...
-CUDA8_GENCODE = -gencode=arch=compute_35,code=sm_35 \
+  ...
-                -gencode=arch=compute_50,code=sm_50 \
+  CUDA8_GENCODE = -gencode=arch=compute_35,code=sm_35 \
-                -gencode=arch=compute_60,code=sm_60 \
+                  -gencode=arch=compute_50,code=sm_50 \
-                -gencode=arch=compute_61,code=sm_61
+                  -gencode=arch=compute_60,code=sm_60 \
-...
+                  -gencode=arch=compute_61,code=sm_61
-```
+  ...

pytorch编译安装过程.1560803470.txt.gz · 最后更改: 2020/02/01 21:40 (外部编辑)