from google.colab import drive
drive.mount('/content/drive/')
import torch
torch.cuda.is_available()
import os
os.chdir('/content/drive/My Drive/AlexNet')
%cd /content/drive/My Drive/AlexNet
!ls
AlexNet是我的云盘目录,只需要切换成自己的目录就可以了。
torch.cuda.is_available() 显示cuda是否可有