我在成功安装CUDA10.2和其对应的CUDNN的情况下,用pip安装了mxnet-cu102,在使用时碰到了如下问题:
代码print(mx.gpu())能成功打印出gpu(0)
但是,
a = nd.array([1, 2, 3], ctx=mx.gpu(0))
print(a)
这段代码会之间陷入无响应状态,既不会输出结果也不会报错,但是GPU监视窗口显示有一个python进程。
以下是我的配置信息:
----------Python Info----------
Version : 3.6.13
Compiler : MSC v.1916 64 bit (AMD64)
Build : (‘default’, ‘Mar 16 2021 11:37:27’)
Arch : (‘64bit’, ‘WindowsPE’)
------------Pip Info-----------
Version : 21.2.2
Directory : G:\anaconda3\envs\gluon\lib\site-packages\pip
----------MXNet Info-----------
Version : 1.7.0
Directory : G:\anaconda3\envs\gluon\lib\site-packages\mxnet
Commit hash file “G:\anaconda3\envs\gluon\lib\site-packages\mxnet\COMMIT_HASH” not found. Not installed from pre-built package or built from source.
Library : [‘G:\anaconda3\envs\gluon\lib\site-packages\mxnet\libmxnet.dll’]
Build features:
CUDA
CUDNN
NCCL
CUDA_RTC
TENSORRT
CPU_SSE
CPU_SSE2
CPU_SSE3
CPU_SSE4_1
CPU_SSE4_2
CPU_SSE4A
CPU_AVX
CPU_AVX2
OPENMP
SSE
F16C
JEMALLOC
BLAS_OPEN
BLAS_ATLAS
BLAS_MKL
BLAS_APPLE
LAPACK
MKLDNN
OPENCV
CAFFE
PROFILER
DIST_KVSTORE
CXX14
INT64_TENSOR_SIZE
SIGNAL_HANDLER
DEBUG
TVM_OP
----------System Info----------
Platform : Windows-10-10.0.22000-SP0
system : Windows
node : txh
release : 10
version : 10.0.22000
----------Hardware Info----------
machine : AMD64
processor : Intel64 Family 6 Model 141 Stepping 1, GenuineIntel
Name
11th Gen Intel® Core™ i7-11800H @ 2.30GHz