在谷歌colab上使用演示一下BlendMask应用
在本地装载 Google 云端硬盘
以下示例展示了如何使用授权代码在您的运行时上装载 Google 云端硬盘,以及如何在那里写入和读取文件。
from google.colab import drive
drive.mount('/content/drive')
会弹出一个链接,点开登录谷歌账户得到授权码。
%cd /content/drive/MyDrive/PyTorch/
%ls
用% cd到要操作的目录下。
安装PyTorch
! /opt/bin/nvidia-smi
! nvcc --version
用!号是使用系统命令,查看显卡版本和cuda版本号。
我这里cuda版本是10.1。所以安装PyTorch的时候用对应的版本1。
! pip install torch==1.7.1+cu101 torchvision==0.8.2