pytorch 安装
安装显卡驱动 输入nvidia-smi，查看cuda version支持的最高版本。输入nvcc -V查看本机安装的cuda 去pytorch官网，安装指定版本的pytorch。 查看是否安装成功：impot torch print(torch.cuda.is_available())
安装显卡驱动 输入nvidia-smi，查看cuda version支持的最高版本。输入nvcc -V查看本机安装的cuda 去pytorch官网，安装指定版本的pytorch。 查看是否安装成功：impot torch print(torch.cuda.is_available())