本文介绍: 目前网络上根据电脑配置安装合适的深度学习环境帖子已经很多了,但是现实中会出现很久之前已经安装对应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

原文地址:https://blog.csdn.net/woshicaiji12138/article/details/134796949

本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任

如若转载,请注明出处:http://www.7code.cn/show_47294.html

如若内容造成侵权/违法违规/事实不符,请联系代码007邮箱suwngjj01@126.com进行投诉反馈,一经查实,立即删除

发表回复

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