Closed LiFXe closed 3 years ago
panzheyi
commented
3 years ago Hi, it seems that the beta version of mxnet is removed on pip.
Instead, you can find a related version on https://repo.mxnet.io/dist/index.html (as the API of beta version is not stable, please download the version with the nearest built date to 20190108).
LiFXe
commented
3 years ago 谢谢，请问cuda与cudnn版本是？现在会报错
Successfully loading the model st-metanet [epoch: 131]
seq2seq_ (
Parameter seq2seq_encoder_c0_gru0_i2h_weight (shape=(192, 3), dtype=<class 'numpy.float32'>)
Parameter seq2seq_encoder_c0_gru0_h2h_weight (shape=(192, 64), dtype=<class 'numpy.float32'>)
Parameter seq2seq_encoder_c0_gru0_i2h_bias (shape=(192,), dtype=<class 'numpy.float32'>)
Parameter seq2seq_encoder_c0_gru0_h2h_bias (shape=(192,), dtype=<class 'numpy.float32'>)
Parameter seq2seq_encoder_c1_dense_z_w_dense0_weight (shape=(16, 32), dtype=float32)
Parameter seq2seq_encoder_c1_dense_z_w_dense0_bias (shape=(16,), dtype=float32)
Parameter seq2seq_encoder_c1_dense_z_w_dense1_weight (shape=(2, 16), dtype=float32)
Parameter seq2seq_encoder_c1_dense_z_w_dense1_bias (shape=(2,), dtype=float32)
Parameter seq2seq_encoder_c1_dense_z_w_dense2_weight (shape=(8192, 2), dtype=float32)
Parameter seq2seq_encoder_c1_dense_z_w_dense2_bias (shape=(8192,), dtype=float32)
Parameter seq2seq_encoder_c1_dense_z_b_dense0_weight (shape=(16, 32), dtype=float32)
Parameter seq2seq_encoder_c1_dense_z_b_dense0_bias (shape=(16,), dtype=float32)
Parameter seq2seq_encoder_c1_dense_z_b_dense1_weight (shape=(2, 16), dtype=float32)
Parameter seq2seq_encoder_c1_dense_z_b_dense1_bias (shape=(2,), dtype=float32)
Parameter seq2seq_encoder_c1_dense_z_b_dense2_weight (shape=(1, 2), dtype=float32)
Parameter seq2seq_encoder_c1_dense_z_b_dense2_bias (shape=(1,), dtype=float32)
Parameter seq2seq_encoder_c1_dense_r_w_dense0_weight (shape=(16, 32), dtype=float32)
Parameter seq2seq_encoder_c1_dense_r_w_dense0_bias (shape=(16,), dtype=float32)
Parameter seq2seq_encoder_c1_dense_r_w_dense1_weight (shape=(2, 16), dtype=float32)
Parameter seq2seq_encoder_c1_dense_r_w_dense1_bias (shape=(2,), dtype=float32)
Parameter seq2seq_encoder_c1_dense_r_w_dense2_weight (shape=(8192, 2), dtype=float32)
Parameter seq2seq_encoder_c1_dense_r_w_dense2_bias (shape=(8192,), dtype=float32)
Parameter seq2seq_encoder_c1_dense_r_b_dense0_weight (shape=(16, 32), dtype=float32)
Parameter seq2seq_encoder_c1_dense_r_b_dense0_bias (shape=(16,), dtype=float32)
Parameter seq2seq_encoder_c1_dense_r_b_dense1_weight (shape=(2, 16), dtype=float32)
Parameter seq2seq_encoder_c1_dense_r_b_dense1_bias (shape=(2,), dtype=float32)
Parameter seq2seq_encoder_c1_dense_r_b_dense2_weight (shape=(1, 2), dtype=float32)
Parameter seq2seq_encoder_c1_dense_r_b_dense2_bias (shape=(1,), dtype=float32)
Parameter seq2seq_encoder_c1_dense_i2h_w_dense0_weight (shape=(16, 32), dtype=float32)
Parameter seq2seq_encoder_c1_dense_i2h_w_dense0_bias (shape=(16,), dtype=float32)
Parameter seq2seq_encoder_c1_dense_i2h_w_dense1_weight (shape=(2, 16), dtype=float32)
Parameter seq2seq_encoder_c1_dense_i2h_w_dense1_bias (shape=(2,), dtype=float32)
Parameter seq2seq_encoder_c1_dense_i2h_w_dense2_weight (shape=(4096, 2), dtype=float32)
Parameter seq2seq_encoder_c1_dense_i2h_w_dense2_bias (shape=(4096,), dtype=float32)
Parameter seq2seq_encoder_c1_dense_i2h_b_dense0_weight (shape=(16, 32), dtype=float32)
Parameter seq2seq_encoder_c1_dense_i2h_b_dense0_bias (shape=(16,), dtype=float32)
Parameter seq2seq_encoder_c1_dense_i2h_b_dense1_weight (shape=(2, 16), dtype=float32)
Parameter seq2seq_encoder_c1_dense_i2h_b_dense1_bias (shape=(2,), dtype=float32)
Parameter seq2seq_encoder_c1_dense_i2h_b_dense2_weight (shape=(1, 2), dtype=float32)
Parameter seq2seq_encoder_c1_dense_i2h_b_dense2_bias (shape=(1,), dtype=float32)
Parameter seq2seq_encoder_c1_dense_h2h_w_dense0_weight (shape=(16, 32), dtype=float32)
Parameter seq2seq_encoder_c1_dense_h2h_w_dense0_bias (shape=(16,), dtype=float32)
Parameter seq2seq_encoder_c1_dense_h2h_w_dense1_weight (shape=(2, 16), dtype=float32)
Parameter seq2seq_encoder_c1_dense_h2h_w_dense1_bias (shape=(2,), dtype=float32)
Parameter seq2seq_encoder_c1_dense_h2h_w_dense2_weight (shape=(4096, 2), dtype=float32)
Parameter seq2seq_encoder_c1_dense_h2h_w_dense2_bias (shape=(4096,), dtype=float32)
Parameter seq2seq_encoder_c1_dense_h2h_b_dense0_weight (shape=(16, 32), dtype=float32)
Parameter seq2seq_encoder_c1_dense_h2h_b_dense0_bias (shape=(16,), dtype=float32)
Parameter seq2seq_encoder_c1_dense_h2h_b_dense1_weight (shape=(2, 16), dtype=float32)
Parameter seq2seq_encoder_c1_dense_h2h_b_dense1_bias (shape=(2,), dtype=float32)
Parameter seq2seq_encoder_c1_dense_h2h_b_dense2_weight (shape=(1, 2), dtype=float32)
Parameter seq2seq_encoder_c1_dense_h2h_b_dense2_bias (shape=(1,), dtype=float32)
Parameter seq2seq_encoder_g0_graph_weight (shape=(1, 1), dtype=<class 'numpy.float32'>)
Parameter seq2seq_encoder_g0_graph_mlp0_dense0_weight (shape=(16, 96), dtype=float32)
Parameter seq2seq_encoder_g0_graph_mlp0_dense0_bias (shape=(16,), dtype=float32)
Parameter seq2seq_encoder_g0_graph_mlp0_dense1_weight (shape=(2, 16), dtype=float32)
Parameter seq2seq_encoder_g0_graph_mlp0_dense1_bias (shape=(2,), dtype=float32)
Parameter seq2seq_encoder_g0_graph_mlp0_dense2_weight (shape=(8192, 2), dtype=float32)
Parameter seq2seq_encoder_g0_graph_mlp0_dense2_bias (shape=(8192,), dtype=float32)
Parameter seq2seq_decoder_c0_gru0_i2h_weight (shape=(192, 3), dtype=<class 'numpy.float32'>)
Parameter seq2seq_decoder_c0_gru0_h2h_weight (shape=(192, 64), dtype=<class 'numpy.float32'>)
Parameter seq2seq_decoder_c0_gru0_i2h_bias (shape=(192,), dtype=<class 'numpy.float32'>)
Parameter seq2seq_decoder_c0_gru0_h2h_bias (shape=(192,), dtype=<class 'numpy.float32'>)
Parameter seq2seq_decoder_c1_dense_z_w_dense0_weight (shape=(16, 32), dtype=float32)
Parameter seq2seq_decoder_c1_dense_z_w_dense0_bias (shape=(16,), dtype=float32)
Parameter seq2seq_decoder_c1_dense_z_w_dense1_weight (shape=(2, 16), dtype=float32)
Parameter seq2seq_decoder_c1_dense_z_w_dense1_bias (shape=(2,), dtype=float32)
Parameter seq2seq_decoder_c1_dense_z_w_dense2_weight (shape=(8192, 2), dtype=float32)
Parameter seq2seq_decoder_c1_dense_z_w_dense2_bias (shape=(8192,), dtype=float32)
Parameter seq2seq_decoder_c1_dense_z_b_dense0_weight (shape=(16, 32), dtype=float32)
Parameter seq2seq_decoder_c1_dense_z_b_dense0_bias (shape=(16,), dtype=float32)
Parameter seq2seq_decoder_c1_dense_z_b_dense1_weight (shape=(2, 16), dtype=float32)
Parameter seq2seq_decoder_c1_dense_z_b_dense1_bias (shape=(2,), dtype=float32)
Parameter seq2seq_decoder_c1_dense_z_b_dense2_weight (shape=(1, 2), dtype=float32)
Parameter seq2seq_decoder_c1_dense_z_b_dense2_bias (shape=(1,), dtype=float32)
Parameter seq2seq_decoder_c1_dense_r_w_dense0_weight (shape=(16, 32), dtype=float32)
Parameter seq2seq_decoder_c1_dense_r_w_dense0_bias (shape=(16,), dtype=float32)
Parameter seq2seq_decoder_c1_dense_r_w_dense1_weight (shape=(2, 16), dtype=float32)
Parameter seq2seq_decoder_c1_dense_r_w_dense1_bias (shape=(2,), dtype=float32)
Parameter seq2seq_decoder_c1_dense_r_w_dense2_weight (shape=(8192, 2), dtype=float32)
Parameter seq2seq_decoder_c1_dense_r_w_dense2_bias (shape=(8192,), dtype=float32)
Parameter seq2seq_decoder_c1_dense_r_b_dense0_weight (shape=(16, 32), dtype=float32)
Parameter seq2seq_decoder_c1_dense_r_b_dense0_bias (shape=(16,), dtype=float32)
Parameter seq2seq_decoder_c1_dense_r_b_dense1_weight (shape=(2, 16), dtype=float32)
Parameter seq2seq_decoder_c1_dense_r_b_dense1_bias (shape=(2,), dtype=float32)
Parameter seq2seq_decoder_c1_dense_r_b_dense2_weight (shape=(1, 2), dtype=float32)
Parameter seq2seq_decoder_c1_dense_r_b_dense2_bias (shape=(1,), dtype=float32)
Parameter seq2seq_decoder_c1_dense_i2h_w_dense0_weight (shape=(16, 32), dtype=float32)
Parameter seq2seq_decoder_c1_dense_i2h_w_dense0_bias (shape=(16,), dtype=float32)
Parameter seq2seq_decoder_c1_dense_i2h_w_dense1_weight (shape=(2, 16), dtype=float32)
Parameter seq2seq_decoder_c1_dense_i2h_w_dense1_bias (shape=(2,), dtype=float32)
Parameter seq2seq_decoder_c1_dense_i2h_w_dense2_weight (shape=(4096, 2), dtype=float32)
Parameter seq2seq_decoder_c1_dense_i2h_w_dense2_bias (shape=(4096,), dtype=float32)
Parameter seq2seq_decoder_c1_dense_i2h_b_dense0_weight (shape=(16, 32), dtype=float32)
Parameter seq2seq_decoder_c1_dense_i2h_b_dense0_bias (shape=(16,), dtype=float32)
Parameter seq2seq_decoder_c1_dense_i2h_b_dense1_weight (shape=(2, 16), dtype=float32)
Parameter seq2seq_decoder_c1_dense_i2h_b_dense1_bias (shape=(2,), dtype=float32)
Parameter seq2seq_decoder_c1_dense_i2h_b_dense2_weight (shape=(1, 2), dtype=float32)
Parameter seq2seq_decoder_c1_dense_i2h_b_dense2_bias (shape=(1,), dtype=float32)
Parameter seq2seq_decoder_c1_dense_h2h_w_dense0_weight (shape=(16, 32), dtype=float32)
Parameter seq2seq_decoder_c1_dense_h2h_w_dense0_bias (shape=(16,), dtype=float32)
Parameter seq2seq_decoder_c1_dense_h2h_w_dense1_weight (shape=(2, 16), dtype=float32)
Parameter seq2seq_decoder_c1_dense_h2h_w_dense1_bias (shape=(2,), dtype=float32)
Parameter seq2seq_decoder_c1_dense_h2h_w_dense2_weight (shape=(4096, 2), dtype=float32)
Parameter seq2seq_decoder_c1_dense_h2h_w_dense2_bias (shape=(4096,), dtype=float32)
Parameter seq2seq_decoder_c1_dense_h2h_b_dense0_weight (shape=(16, 32), dtype=float32)
Parameter seq2seq_decoder_c1_dense_h2h_b_dense0_bias (shape=(16,), dtype=float32)
Parameter seq2seq_decoder_c1_dense_h2h_b_dense1_weight (shape=(2, 16), dtype=float32)
Parameter seq2seq_decoder_c1_dense_h2h_b_dense1_bias (shape=(2,), dtype=float32)
Parameter seq2seq_decoder_c1_dense_h2h_b_dense2_weight (shape=(1, 2), dtype=float32)
Parameter seq2seq_decoder_c1_dense_h2h_b_dense2_bias (shape=(1,), dtype=float32)
Parameter seq2seq_decoder_g0_graph_weight (shape=(1, 1), dtype=<class 'numpy.float32'>)
Parameter seq2seq_decoder_g0_graph_mlp0_dense0_weight (shape=(16, 96), dtype=float32)
Parameter seq2seq_decoder_g0_graph_mlp0_dense0_bias (shape=(16,), dtype=float32)
Parameter seq2seq_decoder_g0_graph_mlp0_dense1_weight (shape=(2, 16), dtype=float32)
Parameter seq2seq_decoder_g0_graph_mlp0_dense1_bias (shape=(2,), dtype=float32)
Parameter seq2seq_decoder_g0_graph_mlp0_dense2_weight (shape=(8192, 2), dtype=float32)
Parameter seq2seq_decoder_g0_graph_mlp0_dense2_bias (shape=(8192,), dtype=float32)
Parameter decoder0_proj_weight (shape=(2, 96), dtype=float32)
Parameter decoder0_proj_bias (shape=(2,), dtype=float32)
Parameter geo_encoder_dense0_weight (shape=(32, 989), dtype=float32)
Parameter geo_encoder_dense0_bias (shape=(32,), dtype=float32)
Parameter geo_encoder_dense1_weight (shape=(32, 32), dtype=float32)
Parameter geo_encoder_dense1_bias (shape=(32,), dtype=float32)
)
NUMBER OF PARAMS: 268224
INFO:root:Processing 1000 timestamps
INFO:root:Processing 2000 timestamps
INFO:root:shape of feature: (2266, 1024, 989)
INFO:root:shape of data: (2266, 12, 1024, 3)
INFO:root:shape of label: (2266, 3, 1024, 3)
INFO:root:Processing 0 timestamps
INFO:root:shape of feature: (346, 1024, 989)
INFO:root:shape of data: (346, 12, 1024, 3)
INFO:root:shape of label: (346, 3, 1024, 3)
INFO:root:Processing 0 timestamps
INFO:root:shape of feature: (306, 1024, 989)
INFO:root:shape of data: (306, 12, 1024, 3)
INFO:root:shape of label: (306, 3, 1024, 3)
Traceback (most recent call last):
File "train.py", line 172, in
Stack trace returned 10 entries:
[bt] (0) /home/lfx19/anaconda3/envs/stmetanet_lfx/lib/python3.5/site-packages/mxnet/libmxnet.so(+0x40585a) [0x7f270f00f85a]
[bt] (1) /home/lfx19/anaconda3/envs/stmetanet_lfx/lib/python3.5/site-packages/mxnet/libmxnet.so(+0x405e71) [0x7f270f00fe71]
[bt] (2) /home/lfx19/anaconda3/envs/stmetanet_lfx/lib/python3.5/site-packages/mxnet/libmxnet.so(+0x3557f7b) [0x7f2712161f7b]
[bt] (3) /home/lfx19/anaconda3/envs/stmetanet_lfx/lib/python3.5/site-packages/mxnet/libmxnet.so(+0x37a637f) [0x7f27123b037f]
[bt] (4) /home/lfx19/anaconda3/envs/stmetanet_lfx/lib/python3.5/site-packages/mxnet/libmxnet.so(+0x37ab979) [0x7f27123b5979]
[bt] (5) /home/lfx19/anaconda3/envs/stmetanet_lfx/lib/python3.5/site-packages/mxnet/libmxnet.so(mxnet::imperative::PushFCompute(std::function<void (nnvm::NodeAttrs const&, mxnet::OpContext const&, std::vector<mxnet::TBlob, std::allocator
LiFXe
commented
3 years ago 非常感谢！另现在又遇新的报错，希望能得到您的指点，谢谢🙏您好。pip上现在看不到过去版本的mxnet了。https://dist.mxnet.io/python/cu90
—You are receiving this because you authored the thread.Reply to this email directly, view it on GitHub, or unsubscribe.
[
{
"@context": "http://schema.org",
"@type": "EmailMessage",
"potentialAction": {
"@type": "ViewAction",
"target": "https://github.com/panzheyi/ST-MetaNet/issues/5#issuecomment-692545539",
"url": "https://github.com/panzheyi/ST-MetaNet/issues/5#issuecomment-692545539",
"name": "View Issue"
},
"description": "View this Issue on GitHub",
"publisher": {
"@type": "Organization",
"name": "GitHub",
"url": "https://github.com"
}
}
]
panzheyi
commented
3 years ago cuda与cudnn版本没有特别的要求，只要cuda版本和mxnet-cu的版本匹配，cudnn版本与cuda版本匹配即可。
作者您好，请问如何装此版本？pip命令报错找不到此版本。期待您的回复，提前感谢！