Pytorch学习笔记 - Pytorch安装
前言
最近在学习Pytorch,希望通过写博客的方式,一方面记录下自己在学习过程中遇到的坑,方便自己以后查阅,另一方面也希望能够将自己的学习笔记分享出来,给遇到类似问题的人一些帮助。
开发环境准备
本机环境:Ubuntu16.04 / GTX 1660 Ti
在下载相应的安装包之前,到torch官方网站查询支持的cuda版本,如下图,选择安装cuda11.1
(注:根据在GTX 3060 Ti显卡上的测试,30系的显卡貌似不支持cuda10系,所以我在另一台3060 Ti的电脑上安装的是cuda11.1)
确定好了cuda的版本之后,再到CUDA Toolkit Documentation去查询适配cuda版本的显卡驱动版本