Tensorflow与cuda版本关系(附加多个cuda版本安装)

Tensorflow与cuda版本关系(附加多个cuda版本安装)

在这里插入图片描述

多说一句

1.windows

如果,在网上down的代码用的tf的版本与你本机cuda不一致的话,可以在本机安装多个cuda版本,到时候再根据你自己的项目来选择使用哪个版本的cuda(就我本机win10为栗子)
在这里插入图片描述
1.安装cuda,这一步在网上太多教程,随便找个就行(cuda安装教程(windows)
2.一般来说,如果cuda默认安装,路径应该与我上面贴出来的图中的路径一致:C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA
3.由于我安装了多个版本cuda,所以在该文件夹中可以看到一共有两个文件夹,每个文件夹的名字就是安装cuda的版本号
4.如果是默认安装的话,cuda应该会自动帮你把cuda添加到环境变量中,打开环境变量就能看到
在这里插入图片描述
环境变量中一共有两个cuda版本,如果你想使用10.1的话只需要把10.1的路径放在10.0的前面,这样系统就会自动先识别出10.1版本的cuda
5.最后,你可以使用nvcc --version去看看cuda是不是变了(cmd最好先关闭再打开,不然可能还是会显示旧版本的cuda版本号)

2.linux

linux如果同时安装了两个cuda版本,也是通过环境变量进行设置。如下图,我在我的电脑上面安装了两个版本的cuda。(一个版本是10.1,还有一个版本是11.1,其实还有一个系统自带的cuda,具体版本我也不清楚)
在这里插入图片描述

1.sudo vim ~/.bashrc 打开环境变量
2.在这里插入图片描述在环境变量中通过设置路径,其中/usr/local是cuda的安装路径,如果在安装的时候不设置路径,系统默认路径应该跟我一样。其中cuda-10.1就是你想使用哪个版本的cuda了,如果这里我想使用cuda-11.1版本,那我就把cuda-10.1修改为11.1,其他的都不用变。
3.保存退出vim编辑器,然后source ~/.bashrc更新下就好了。
4.最后,还是使用nvcc --version去看看cuda是不是变了
在这里插入图片描述

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

进我的收藏吃灰吧~~

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值