cuda环境配置,tf,torch,gpu配置
最近有很多同学准备入门深度学习,为了做软件课设。我在这里分享一些我的经验。
首先确定你的电脑有NVIDIA显卡,目前只有n卡能提供gpu算力。
查看你的电脑支持的最高cuda版本
打卡英伟达控制面板
或者直接在搜索栏中搜索
然后就会看到以下的画面
GeForce GTX 1050 是我的n卡型号,注意记下上面的版本数值,这个数值是Driver Version(驱动程序版本)。我们要根据这个驱动程序版本去选择对应的cuda版本。
这是我在官网截的图,驱动版本对应关系
注意版本是 “向下兼容”
就是说, 比如我的457.49,在上面表格中 可以看到CUDA 11.1.1 Update 1对应的>=456.81(window下)。向下兼容就是我们只能装小于等于这个版本的
选择cud