|
发表于 2020-10-29 20:03:49
4414 浏览 8 回复
深度强化学习算法环境配置
3080显卡,安装完455.23的显卡驱动,可以使用nvidia-smi命令查看,但是运行tensorflow1.8.0版本的代码时,无法使用GPU,不知道该如何匹配cuda和cudnn版本
jario已获得悬赏 1 阿木币最佳答案
系统是ubuntu1804吗,cuda和cudnn需要匹配tensorflow的版本,看tensorflow的安装说明里有介绍
|
扫一扫浏览分享
|
|
|
|
|
|
|
发表于 2020-10-29 20:22:43
系统是ubuntu1804吗,cuda和cudnn需要匹配tensorflow的版本,看tensorflow的安装说明里有介绍 |
|
|
|
|
|
|
|
楼主|
发表于 2020-10-29 20:40:16
我在网上看过,我想知道具体怎么安装的,我之前安装了cuda10.1,由于版本匹配的问题,最后重装了系统 |
|
|
|
|
|
|
|
楼主|
发表于 2020-10-29 20:48:02
我在网上看过,我想知道具体怎么安装的,我之前安装了cuda10.1,由于版本匹配的问题,最后重装了系统 |
|
|
|
|
|
|
|
发表于 2020-10-29 20:59:37
你是ubuntu1804吗,nvidia官网有教程,下载deb文件,然后sudo dpkg -i xxx.deb就可以直接安装 |
|
|
|
|
|
|
|
发表于 2020-10-29 21:00:52
你是ubuntu1804吗,nvidia官网有教程,下载deb文件,然后sudo dpkg -i xxx.deb就可以直接安装 |
|
|
|
|
|
|
|
楼主|
发表于 2020-10-29 21:02:08
这种方式安装的话,如果我换一个版本的tensorflow就不能使用了 |
|
|
|
|
|
|
|
楼主|
发表于 2020-10-29 21:04:22
我是ubuntu18.04。我直接使用conda虚拟环境来安tf,如果安1.8版本的tf直接报错 |
|
|
|
|
|
|
|
发表于 2020-10-29 21:31:50
我们一直是这样用的,就是保证tf或pytorch与cuda、cudnn版本对应,没必要一直换版本吧。如果需要新特性的话,一段时间进行一次总体升级。 |
|
|
|
|
|
|
|