1.指定工作文件夹(运行可以相对路径!)

# 指定当前的工作文件夹
import os

# 此处为google drive中的文件路径,drive为之前指定的工作根目录,要加上
os.chdir("/content/drive/My Drive/app/WDSR_pytorch") 

 

2.查询版本

!nvcc --version #查看cuda版本
!lsb_release -a #查看系统版本

 

相关文章: