pytorch使用教程

162 阅读1分钟
  1. 安装英伟达驱动 驱动下载地址:www.nvidia.cn/Download/in…

0880c888051082d4d303.png

image.png

0880c888051082948206.png

  1. 安装 anaconda docs.anaconda.com/free/anacon…
  1. 安装 cuda 的官网
  • 先知道自己要安装什么版本的cuda

  • 然后安装cuda image.png

  • 测试是否安装成功 /usr/local/cuda/bin/nvcc -V

  1. 安装 pytorch 官网
  • 在你虚拟的conda环境里执行 image.png

参考: