File tree Expand file tree Collapse file tree 3 files changed +4
-5
lines changed Expand file tree Collapse file tree 3 files changed +4
-5
lines changed Original file line number Diff line number Diff line change 33# # Set up Cuda-related environment settings
44
55while [ " $TF_NEED_CUDA " == " " ]; do
6- read -p " Do you wish to bulid TensorFlow with GPU support? [y/n] " INPUT
6+ read -p " Do you wish to build TensorFlow with GPU support? [y/n] " INPUT
77 case $INPUT in
88 [Yy]* ) echo -e " GPU support will be enabled for TensorFlow\n" ; TF_NEED_CUDA=1;;
99 [Nn]* ) echo -e " No GPU support will be enabled for TensorFlow\n" ; TF_NEED_CUDA=0;;
Original file line number Diff line number Diff line change 66Additional details about the TensorFlow programming model and the underlying
77implementation can be found in out white paper:
88
9- * [ TensorFlow: Large-scale machine learning on heterogeneous systems] ( http://www .tensorflow.org/whitepaper2015.pdf )
9+ * [ TensorFlow: Large-scale machine learning on heterogeneous systems] ( http://download .tensorflow.org/paper /whitepaper2015.pdf )
1010
1111### Citation <a class =" md-anchor " id =" AUTOGENERATED-citation " ></a >
1212
Original file line number Diff line number Diff line change @@ -179,14 +179,13 @@ bazel build -c opt tensorflow/models/rnn/ptb:ptb_word_lm
179179And if you have a fast GPU, run the following:
180180
181181```
182- bazel build -c opt tensorflow --config=cuda \
183- tensorflow/models/rnn/ptb:ptb_word_lm
182+ bazel build -c opt --config=cuda tensorflow/models/rnn/ptb:ptb_word_lm
184183```
185184
186185Now we can run the model:
187186
188187```
189- bazel-bin/... /ptb_word_lm \
188+ bazel-bin/tensorflow/models/rnn/ptb /ptb_word_lm \
190189 --data_path=/tmp/simple-examples/data/ --alsologtostderr --model small
191190```
192191
You can’t perform that action at this time.
0 commit comments