@@ -90,10 +90,34 @@ ImportError: DLL load failed: The specified procedure could not be found.
90
90
You can download it and put it in the PyTorch directory or use it in a standalone way.
91
91
There're more details about MSVC 2017 setup in [ #23 ] ( https://github.com/peterjc123/pytorch-scripts/issues/23 ) .
92
92
``` Powershell
93
+ ################################################################################
94
+ # These are the flags that you can set before running the scripts.
95
+ ################################################################################
96
+
93
97
# You can specify which version you want to build
94
98
# If you omit it, it will build for the master branch on default
95
99
set PYTORCH_BUILD_VERSION=0.4.1
96
100
101
+ # If you want to build your own fork of PyTorch
102
+ set PYTORCH_REPO=pytorch
103
+
104
+ # If you want to build for a specific branch
105
+ # Note: If you specify PYTORCH_BUILD_VERSION too, it won't be used.
106
+ set PYTORCH_BRANCH=master
107
+
108
+ # If you want to skip the confirmation of the flags configured by our script
109
+ set NO_PROMPT=1
110
+
111
+ # If you want to continue build without a verified VS version (Safe for CPU builds)
112
+ set SKIP_VS_VER_CHECK=1
113
+
114
+ # If you want to skip prompt for building for Python 2.7 (Not recommended)
115
+ set SKIP_PY_VER_CHECK=1
116
+
117
+ ################################################################################
118
+ # These are the available scripts that you can use.
119
+ ################################################################################
120
+
97
121
# If you don't want to override the default settings
98
122
auto.bat
99
123
@@ -109,6 +133,12 @@ cuda90.bat
109
133
# If you want to compile with CUDA 9.1
110
134
cuda91.bat
111
135
136
+ # If you want to compile with CUDA 9.2
137
+ cuda92.bat
138
+
139
+ # If you want to compile with CUDA 10.0
140
+ cuda100.bat
141
+
112
142
```
113
143
114
144
# Using Examples
0 commit comments