Pytorch框架安装笔记(重点)

一、安装过程

1. 查看本机电脑的 GPU驱动,看看当前 驱动版本可以 支持的cuda版本,对应关系如下:
Pytorch框架安装笔记(重点)_第1张图片
最新可查阅官方文档

注:驱动是向下兼容的,其决定了可安装的CUDA Toolkit的最高版本

2.由1中确定cuda版本以后, 以cuda为中心,寻找配件进行安装,根据cuda版本确定cudnn版本

3. 由cuda确定Pytorch的版本
Pytorch框架安装笔记(重点)_第2张图片

注:虽有的卡驱动更新至较新版本,且CUDA Toolkit及PyTorc

你可能感兴趣的:(DL,Pytorch,pytorch,深度学习,python)