导读 最近入手了一台搭载GTX1060MQ显卡的笔记本,准备搭建深度学习环境,但折腾了好几天才搞定,特此记录下安装过程,希望能帮到有同样需求的朋...
最近入手了一台搭载GTX1060MQ显卡的笔记本,准备搭建深度学习环境,但折腾了好几天才搞定,特此记录下安装过程,希望能帮到有同样需求的朋友!💪
首先,确保你的Windows 10系统是最新的版本,这能避免后续安装中出现兼容性问题。接着,我们需要正确安装显卡驱动,推荐从NVIDIA官网下载最新版GTX1060MQ驱动,切勿使用过时或不完整的驱动包哦!⚙️
安装完驱动后,就可以着手配置TensorFlow-GPU了。记得提前检查CUDA和cuDNN的版本是否匹配,建议直接安装CUDA 10.0+以及cuDNN 7.4+,这样能最大程度减少兼容性问题。pip安装命令如下:
`pip install tensorflow-gpu==1.13.1`
最后,运行测试代码验证GPU是否正常工作:
```python
import tensorflow as tf
tf.test.is_gpu_available()
```
看到返回True,说明一切OK!🎉
希望这篇笔记对大家有所帮助,如果有任何问题欢迎留言交流~💬