使用Colab运行pytorch文件

#1. 查看pytorch版本：
import torch
print(torch.__version__)
#2.查看cuda版本：
print(torch.version.cuda)


from google.colab import drive
drive.mount('/content/drive/')

#!pip install torch==1.10.2 torchvision==0.11.3 -f https://download.pytorch.org/whl/torch_stable.html
!pip install transformers==3.4.0
!pip install numpy==1.19

!python "/content/drive/MyDrive/model/nlp/bert_model.py"