File tree Expand file tree Collapse file tree 2 files changed +4
-4
lines changed Expand file tree Collapse file tree 2 files changed +4
-4
lines changed Original file line number Diff line number Diff line change 64
64
while [[ " $TF_CUDA_VERSION " == " " ]]; do
65
65
read -p " Are you building against TensorFlow 2.1(including RCs) or newer?[Y/n] " INPUT
66
66
case $INPUT in
67
- [Yy]* ) echo " Build against TensorFlow 2.1 or newer." ; TF_CUDA_VERSION=11 ;;
67
+ [Yy]* ) echo " Build against TensorFlow 2.1 or newer." ; TF_CUDA_VERSION=10.1 ;;
68
68
[Nn]* ) echo " Build against TensorFlow <2.1." ; TF_CUDA_VERSION=10.0;;
69
- " " ) echo " Build against TensorFlow 2.1 or newer." ; TF_CUDA_VERSION=11 ;;
69
+ " " ) echo " Build against TensorFlow 2.1 or newer." ; TF_CUDA_VERSION=10.1 ;;
70
70
* ) echo " Invalid selection: " $INPUT ;;
71
71
esac
72
72
done
Original file line number Diff line number Diff line change 13
13
# See the License for the specific language governing permissions and
14
14
# limitations under the License.
15
15
# ==============================================================================
16
- wget https://github.com/bazelbuild/bazel/releases/download/5.3 .0/bazel_5.3 .0-linux-x86_64.deb
17
- sudo dpkg -i bazel_5.3 .0-linux-x86_64.deb
16
+ wget https://github.com/bazelbuild/bazel/releases/download/5.1 .0/bazel_5.1 .0-linux-x86_64.deb
17
+ sudo dpkg -i bazel_5.1 .0-linux-x86_64.deb
18
18
pip install --upgrade pip setuptools wheel
19
19
pip install -r requirements.txt
You can’t perform that action at this time.
0 commit comments