动手学深度学习_安装过程中的所有坑
·
在Colab中导入自定义的包
参考:http://t.csdn.cn/WAWAu
我的google drive:
在Colab对应的Jupyter的代码块中输入:
import sys
from google.colab import drive
drive.mount('/content/drive')
sys.path.append('/content/drive/My Drive/Colab Notebooks')
于是有:
在Colab中使用GPU
在Edit这一栏的Notebook Settings里面
查看是否设置成功,在代码块中输入:
1.导入相关包
from mxnet import np,npx
# 安装GPU版本
import mxnet as mx
from mxnet import nd,gpu,gluon,autograd
from mxnet.gluon import nn
from mxnet.gluon.data.vision import datasets,transforms
import time
2.测试
ctx = mx.gpu()
ctx

魔乐社区(Modelers.cn) 是一个中立、公益的人工智能社区,提供人工智能工具、模型、数据的托管、展示与应用协同服务,为人工智能开发及爱好者搭建开放的学习交流平台。社区通过理事会方式运作,由全产业链共同建设、共同运营、共同享有,推动国产AI生态繁荣发展。
更多推荐


所有评论(0)