pytorch 安装

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

发表评论

邮箱地址不会被公开。 必填项已用*标注