在windows10上配置深度学习GPU开发环境

说明

其实整体挺简单的,GPU与CPU不同的地方就是,需要先安装cuda和cudnn;安装cuda时需要首先确定自己的显卡需要安什么版本的cuda;安装pytorch的时候,需要确定你的cuda安的是什么版本的,然后选择与之对应的pytorch版本,如果pytorch没有对应的cuda版本那你就卸了重安cuda

建议各软件不安装最新版,这样其他的软件容易没有与之对应的版本

个人环境

系统:windows10

显卡:3090

cuda: 11.1

cudnn: 8.2.2

pytorch: 1.8.0

tensorflow: 2.5.0

安装CUDA

下载地址:

最新版:

https://developer.nvidia.com/cuda-downloads

旧版:

https://developer.nvidia.com/cuda-toolkit-archive

参考:

https://zhuanlan.zhihu.com/p/94220564?utm_source=wechat_session

查看已安装cuda版本:

 

如果没有安装cuda,会显示

吐槽:

查看已安装cuda版本时,有的博客显示用这种方式查看cuda版本(NVIDIA控制面板,左下系统信息,组件)。

真坑呀这个位置显示的是当前显卡所能支持的、最高的cuda版本,不是已安装的cuda版本,弄得我还以为cuda安装好了呢,开始按pytorch,最后报错,缺少cudatoolkit包

安装pytorch

安装tensorflow

安装opencv

参考:

https://mp.csdn.net/mp_blog/creation/editor/111470521

这个位置没有啥难的,就是版本不同,一定要根据实际情况选好版本

实在是安装的时候没遇到什么问题,所以没啥可写的,就是使用anaconda创建了个虚拟的环境python3.6,在这个环境上安装pytorch包和tensorflow包,个人之所以安装tensorflow,主要还是因为使用tensorboard

猜你喜欢

转载自blog.csdn.net/XDH19910113/article/details/119991796