本文介绍: 目前网络上根据电脑配置安装合适的深度学习环境帖子已经很多了,但是现实中会出现很久之前已经安装对应cuda,但是现在忘记了当时安装的是什么版本本文针对这一问题展开攻略。

目前网络上根据电脑配置安装合适的深度学习环境帖子已经很多了,但是现实中会出现很久之前已经安装了对应cuda,但是现在忘记了当时安装的是什么版本本文针对这一问题展开攻略。

1 cuda安装版本查询

我们查询自己应该安装什么版本cuda时,一般是:
管理员运行windows powershell然后输入命令

nvidia-smi

在这里插入图片描述

之后根据显示cuda版本选择一个大于显示版本号cuda版本进行安装。因此,我们可以知道nvidiasmi显示出的cuda版本仅仅只是本电脑显卡驱动支持的最高版本cuda。原先电脑上已有的cuda版本要通过命令查询

nvcc --version

在这里插入图片描述

2 虚拟环境创建

安装新的pytorch环境创建一个新的虚拟环境这里使用conda创建

conda create -n py38
conda activate py38

3pytroch安装

pytorch官网
接下来,在上述链接选择对于cuda版本的pytorch安装命令然后anaconda cmd中的对应虚拟环境粘贴运行即可
这里建议使用pip

发表回复

您的邮箱地址不会被公开。 必填项已用 * 标注