在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

在这里插入图片描述

Logo

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

更多推荐