用AIStudio运行自己的Pytorch代码

1.上传自己数据集
将自己的数据集压缩成“数据集名.zip”形式,在AIStudio中找到数据集->创建数据集,然后一步一步按照提示做。
2.创建项目
在AIStudio中找到项目->创建项目,填写项目描述,在点击“创建”前要“添加数据集”,在“个人数据集”中找到自己刚刚上传的数据集,添加了数据集后再点击“创建”。
用AIStudio运行自己的Pytorch代码_第1张图片
也可以在创建好的项目里添加数据集,但此时添加的数据集的大小有限制,所以在创建项目时就选择要添加的数据集更方便。
3.解压数据集
查看自己的项目,点“启动环境”,并选择高级版运行环境:
用AIStudio运行自己的Pytorch代码_第2张图片
,然后进入,在终端执行“unzip 数据集名.zip”。
4.构建Pytorch环境
此时可以查看cuda版本:在终端输入“nvcc -V”,我的是9.2
在终端运行以下命令:
pip install -i https://mirrors.tuna.tsinghua.edu.cn/simple torch1.5.0+cu92 torchvision0.6.0+cu92 -f https://download.pytorch.org/whl/torch_stable.html
回车之后它就在下载了。(早上比晚上下载得快)
下载完成后查看是否可以使用Pytorch,即在终端运行“python”和“import torch”,如果不报错,说明成功了。然后再退出来,终端执行“exit()”。
5.运行Pytorch代码
在文件夹里新建文件“文件名.py”,然后将自己要运行的代码加进去,在终端执行“python 文件名.py”。要注意的是Pytorch代码不能在Notebook里面用。

你可能感兴趣的:(深度学习)