一.Goole Colab
- continuous 12 hours
- free k80 for GPU
- 适合学习、博客、教程
1.点击网址https://colab.research.google.com/notebooks/intro.ipynb#scrollTo=uyGjonhnG02t进入
2.如未登录,注册谷歌账号;如已登录,如下界面。
3.文件->新建python3,或者打开笔记本(之前保存的文件)
4.使用GPU加速的操作:需要设置,修改->笔记本设置,硬件加速器选择GPU。
5.测试是否配置好pytorch的gpu版本
import torch
print(torch.cuda.is_available())
out:True
6.上传文件
如图,点击上传,选择本地文件上传。
二.常用操作指令
1.查看cuda版本
!cat /usr/local/cuda/version.txt
out:CUDA Version 10.1.243
2.卸载torch
!pip uninstall torch
安装gpu版pytorch
!pip install torch torchvision
3.查看pytorch的gpu版本是否安装
import torch
print(torch.cuda.is_available())
4.安装visdom
pip install visdom
from visdom import Visdom