进入网页:
https://developer.nvidia.com/cuda-toolkit-archive
,选择cuda toolkit 10进行下载
点击上图中的连接,进入下载界面,然后依次进行选择
选择好之后,将会进行下载,我们选择的是本地安装,安装包有2g大小。
等下载完成之后,点击下载好的文件进行安装(注意:一定要确认自己的电脑有nvida的显卡),安装软件会弹出下面的界面:
不进行修改,直接点击ok,然后软件会进行解压,如下图:
然后弹出下面系统兼容性检测,等待检测结束。
检测结束之后,会弹出下面的界面,点击同意
然后进入安装模式选择,选择自定义安装
去掉visual studio integration模型前的钩
去掉nvdia geforce experience 前面的钩和display driver前面的钩,然后点击下一步
进入安装路径界面,使用默认,点击下一步
最后等待安装完成
在激活的虚拟环境中,使用下面的命令安装模块
pipenv install --skip-lock numpy pandas matplotlib ipython jupyter
安装tensorflow的cpu和gpu版本,命令如下:
pipenv install --skip-lock tensorflow==2.0.0 tensorflow-gpu==2.0.0
如果直接安装最新的tensorflow版本,不知道什么原因总会出现错误,而安装2.0.0版本则可以正确运行,所以这里选择安装上面的版本。
上面的都安装完成之后,在当前的虚拟环境中,输入
ipython
进入ipython交互环境中,使用代码
import tensorflow as tf
导入tensorflow模块,如果不报错,则证明tensorflow安装成功。
然后验证gpu是否可以使用,输入代码
tf.test.is_gpu_available()
进行测试,输入该代码之后,如果没有问题,则会弹出当前使用的gpu部分信息,并最后输出
True
。
深度学习
用什么
显卡
?3060
显卡
适合
深度学习
吗?都知道
深度学习
很吃
显卡
,显存越大,可以缓存的内容就越多。对于非常吃显存的图像类
深度学习
程序来说,显存太小的
显卡
批处理就不能调太大,否则会程序会报错。......
在使用任何这些库之前,我们需要通过在我们的系统上安装和配置它们来设置我们的 PC 或笔记本电脑。虽然这些库可以安装在不同类型的操作系统上,但今天的讨论仅限于 Windows 操作系统。我还假设您有一台安装了最新版本 Windows 的笔记本电脑或 PC。Keras 和
TensorFlow
是基于
Python
的库。在安装其中任何一个之前,首先,我们需要安装
Python
。有很多方法可以在您的 PC 或笔记本电脑上安装
Python
。因为我们正在执行 ML 和 DL 任务,所以我强烈建议您从 Anacon
二、准备工作
要使用GPU加速,首先就要去安装
CUDA
和cuDNN。 首先 要看一下你的电脑的
显卡
对应的是哪个版本的
CUDA
,千万不要下错了,不然
运行
不了的。 查看方法很多,网上都很容易搜得到,这里介绍其中一种: 在终端输入
nvidia-smi
就可以查看自己的
显卡
对应哪个版本。我电脑的是GTX 1650,对
之前入坑
tensorflow
的时候看着百度上的各种安装教程,装了整整一个星期,看到终端上终于打出“Hello World”,我开心的跟中了奖似的。
之前
tensorflow
的GPU版一直下一直配置不成功,GeForce、
CUDA
一直出现问题,百度了很久也是没有解决方法。然后直接去了anaconda的官网,看了
tensorflow
的安装教程,前面一大堆英文就跳过了,直接看命令。发现就一句命令…
较新版本的
python
3.8/
cuda
11如果想要使用tf1.15,一种方法是重新编译,或者去下载其他人编译好的二进制whl。但是重新编译和找whl都比较麻烦,老黄为了让新
显卡
用户用上tf1.15,和谷歌合作维护了一个tf1.15的库。另一种方法是使用docker,但是docker用起来稍麻烦。
库的地址为:
https://github.com/NVIDIA/
tensorflow
安装命令:
# 推荐在虚拟环境下操作(非必要)
#
python
3 -m virtualenv venv
# sourc