跟着网上教程下载了mxnet-cu101,然后又安装了cudatoolkit 10.1使得测试nd.ndarray([1,2],ctx = mx.gpu())的时候可以正常运行(在第一次的时候卡了好久,后来就正常了)
但是后来又发现在net.initialize(ctx = mx.gpu())的时候很慢,根据我同时nvidia-smi的结果,它一直在往显存里写东西,但写的很慢,大概要写20分钟写500M+这个样子。
是因为cuda版本不对,每次运行都要兼容吗?我没有找到mxnet-cu122
(gpu-mxnet) C:\Users\86182>pip show mxnet-cu101
Name: mxnet-cu101
Version: 1.5.0
Summary: MXNet is an ultra-scalable deep learning framework. This version uses .
Home-page: https://github.com/apache/incubator-mxnet
Author: UNKNOWN
Author-email: UNKNOWN
License: Apache 2.0
Location: d:\miniconda\envs\gpu-mxnet\lib\site-packages
Requires: graphviz, numpy, requests
Required-by:
print(mxnet.version)
1.5.0