cuda环境配置（anaconda虚拟环境版，含pytorch

2024-07-17 02:41| 来源: 网络整理| 查看: 265

cuda环境配置（anaconda虚拟环境版，含pytorch-gpu安装）一、背景二、anaconda虚拟环境创建三、cuda+cudnn安装1.cuda版本查询2.cudatoolkit安装四、pytorch-gpu安装1.安装2.验证pytorch-gpu是否安装成功

一、背景

cuda可以直接装到电脑上，但是安装步骤复杂，且失败率较高。选择anaconda虚拟环境安装，操作简单，且可以结合pytorch、tensorflow等深度学习框架的不同版本，安装不同的版本的cuda，各个虚拟环境互不干扰，也可以一键删除。

二、anaconda虚拟环境创建

anaconda安装配置十分简单可自行查找教程，包括环境变量和镜像源配置。打开cmd命令行（anaconda已放到环境变量）或者Anaconda PowershellPrompt（如下图）：在这里插入图片描述

输入 conda create -n test python=3.8

‘’test‘’为虚拟环境名词，可以自动定义。python版本可以结合自己的环境需求，自行修改，这里用python3.8。在这里插入图片描述

输入y回车，等到运行完毕。输入conda env list，我们可以看到虚拟环境tese已创建完毕，环境名称后面是所在的路径，便于在pycharm等IDE中导入。

在这里插入图片描述环境创建好后，需要进入虚拟环境，输入conda activate test回车：

在这里插入图片描述

可以看到环境由base变为test，表示已经进入test虚拟环境。下一步就是安装cuda和cudnn。

三、cuda+cudnn安装 1.cuda版本查询

打开命令行，输入nvidia-smi 在这里插入图片描述如图，CUDA Version后面的就是电脑显卡可支持的cuda版本，所以我们要装的cuda版本需要

【本文地址】

今日新闻