experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Move default TF builds to cuda 10.

Browse Source 
PiperOrigin-RevId: 227187422
This commit is contained in:
Gunhan Gulsoy 2018-12-28 18:31:41 -08:00 committed by TensorFlower Gardener
parent 817504f1e4
commit 3aaebcbcad
3 changed files with 25 additions and 25 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

42
third_party/toolchains/gpus/cuda/BUILD vendored
View File

File diff suppressed because one or more lines are too long

4
third_party/toolchains/gpus/cuda/cuda/cuda_config.h vendored
View File

@ -19,9 +19,9 @@ limitations under the License.
#define TF_CUDA_CAPABILITIES CudaVersion("3.0")
#define TF_CUDA_VERSION "9.0"
#define TF_CUDA_VERSION "10.0"
#define TF_CUDNN_VERSION "7"
#define TF_CUDA_TOOLKIT_PATH "/usr/local/cuda-9.0"
#define TF_CUDA_TOOLKIT_PATH "/usr/local/cuda-10.0"
#endif // CUDA_CUDA_CONFIG_H_

4
third_party/toolchains/preconfig/ubuntu14.04/gcc-nvcc/clang/bin/crosstool_wrapper_driver_is_not_gcc vendored
View File

@ -49,9 +49,9 @@ import pipes
CPU_COMPILER = ('/usr/bin/gcc')
GCC_HOST_COMPILER_PATH = ('/usr/bin/gcc')
NVCC_PATH = '/usr/local/cuda-9.0/bin/nvcc'
NVCC_PATH = '/usr/local/cuda/bin/nvcc'
PREFIX_DIR = os.path.dirname(GCC_HOST_COMPILER_PATH)
NVCC_VERSION = '9.0'
NVCC_VERSION = '10.0'
def Log(s):
print('gpus/crosstool: {0}'.format(s))