分享web开发知识

注册/登录|最近发布|今日推荐

主页 IT知识网页技术软件开发前端开发代码编程运营维护技术分享教程案例
当前位置:首页 > 运营维护

ubuntu安正mxnet GPU版本

发布时间:2023-09-06 01:51责任编辑:胡小海关键词:暂无标签

安装mxnet GPU
sudo pip install mxnet-cu80==1.1.0

(1)使用系统python验证,命令行中输入:python

1,cpu:

from mxnet import nd
x = nd.array([1,2,3])
x.context
cpu(0)


2,GPU

from mxnet import nd
a = nd.array([1,2,3], ctx=mx.gpu())
a

[ 1. ?2. ?3.]
<NDArray 3 @gpu(0)>,

ubuntu安正mxnet GPU版本

原文地址:https://www.cnblogs.com/adong7639/p/8954476.html

知识推荐

我的编程学习网——分享web前端后端开发技术知识。 垃圾信息处理邮箱 tousu563@163.com 网站地图
icp备案号 闽ICP备2023006418号-8 不良信息举报平台 互联网安全管理备案 Copyright 2023 www.wodecom.cn All Rights Reserved