首页 » 深度学习 » 正文

conda/anaconda一键安装CUDA cuDNN TensorFlow

按照Nvidia官方安装Cuda
按照官方文档安装CUDA配合对应版本的cuDNN会非常繁琐,如果某些地方出错会导致一系列的错误。
利用conda安装tensorflow会自动安装对应版本的CUDA和cuDNN
直接使用conda install tensorflow一键安装,开箱即用。建议在安装之前先创建虚拟环境。
需要注意的是,目前conda支持Python3.7但是不支持Python3.7的TensorFlow。如果在Python3.7环境下安装,conda会自动将Python3.7降级为Python3.6。
还有一点是conda目前支持的CUDA版本是9.2,而目前Nvidia最新的CUDA是10.1,所以如果有最新版本的CUDA需求的话还是得自行安装。

发表评论