win下conda安装多版本cuda

文章目录[隐藏]

需求/问题

电脑只能安装一个cuda,但是往往研究tensorflow-gpu的时候,可能需要多个版本的cuda,利用conda可以实现安装多个版本。

安装教程

查看当前cuda版本

nvcc -V

在这里插入图片描述

查看conda支持的cudatoolkit版本,这个命令就是看下版本支持,看看就行。下面的命令查看cudnn版本比较重要

conda search cudatoolkit --info

在这里插入图片描述

支持cudnn的版本,这里我们安装cudnn7.6.5,cuda10.1,需要的cudatoolkit是10.1

conda search cudnn --info

在这里插入图片描述

创建python环境

conda create -n ten-gpu python=3.7

激活ten-gpu环境

conda activate ten-gpu

在这里插入图片描述
根据上面的提示安装cudatoolkit

conda install cudatoolkit=10.1

在这里插入图片描述

conda install cudnn=7.6.5

在这里插入图片描述
安装gpu版本2.2.0

pip install tensorflow-gpu==2.2.0

在这里插入图片描述

检查是否安装成功

>>> import tensorflow as tf
>>> tf.__version__

在这里插入图片描述

版权声明:本文为CSDN博主「baiqingchun」的原创文章,遵循CC 4.0 BY-SA版权协议,转载请附上原文出处链接及本声明。
原文链接:https://blog.csdn.net/u014196765/article/details/121656137

baiqingchun

我还没有学会写个人说明!

暂无评论

发表评论

相关推荐

使用CUDA+OpenCV加速yolo v4性能

YOLO是You-Only-Look-Once的缩写,它无疑是根据COCO数据集训练的最好的对象检测器之一。YOLOv4是最新的迭代版本,它在准确性和性能之间进行了权衡,使其成为最先进的对象检测器之一。

随笔-Ubuntu中多cuda版本环境配置

随笔-Ubuntu中多cuda版本环境配置 由于每个人做得任务不同,在实验室的服务器上,有多个用户和多个cuda版本,如何配置自己使用哪一个cuda呢? 看一下cuda的安装路径 c