搞了一天时间,总算把pytorch-gpu给配置好了,感觉自己真的太菜了,要努力了!因为废了很大功夫,所以,写下来,方便未来再配置的时候翻阅。
第一,下载cuda安装
参考博客:https://blog.csdn.net/Mind_programmonkey/article/details/99688839
(我怕别人的博客未来被删除,所以自己复制一份)
首先我们要确定本机是否有独立显卡。在计算机-管理-设备管理器-显示适配器中,查看是否有独立显卡,可以看到本机有一个集成显卡和独立显卡NVIDIA GetForce GTX 1050。
第二,下载对应的cuda ,下载地址:https://developer.nvidia.com/cuda-gpus
测试本机独立显卡是否支持CUDA的安装,点击此处查询显卡是否在列表中。从下面第二个图中,可以看到我本机的独立显卡是支持CUDA安装的,计算力是6.1。下载地址为https://developer.nvidia.com/cuda-toolkit-archive