TORCH_VERSION=2.1.0, USE_CUDA=ON, USE_CUDNN=ON, USE_EXCEPTION_PTR=1, USE_GFLAGS=OFF, USE_GLOG=OFF, USE_MKL=ON, USE_MKLDNN=ON, USE_MPI=OFF, USE_NCCL=ON, USE_NNPACK=ON, USE_OPENMP=ON, USE_ROCM=OFF
Searching for GPU management library libnvidia-ml.so 2024/01/13 20:14:03 gpu.go:248: Discovered GPU libraries: [] 2024/01/13 20:14:03 gpu.go:203: Searching for GPU management library librocm_smi64.so 2024/01/13 20:14:03 gpu.go:248: Discovered GPU libraries: [] 2024/01/13 20:...