首页 > 其他分享 >conda环境中cuda的安装

conda环境中cuda的安装

时间:2022-11-09 17:34:22浏览次数:63  
标签:-- torch python conda cuda 安装

今天在conda下d2l环境中,正在使用pytorch学卷积神经网络,想用gpu加速训练,于是使用如下命令

torch.cuda.device_count()

竟然输出0。我是安装过cuda且有gpu的,怎么回事呢。可能是之前没注意版本的原因。
干脆重新安装吧!

环境上的问题

一定要让你用的pip与python对应,否则全都白做
在安装之前,先看好自己的python和pip版本。
我是在conda创造的虚拟环境d2l中进行所有步骤的,其python=3.8,其他环境中也类似。
执行

python -V
python3 -V
pip -V
pip3 -V

(我花了一上午的时间才发现这个问题,用pip安装的torch安装到了python3.9中了,而我用的是python3.8,导致python3.8一直无法使用GPU。最终卸载掉了多余的python,接下来的python和pip都对应的python3.8.13)

接下来是conda的修改镜像环节,目的是加速下载速度。
下面的操作是常用命令:

  1. 查看镜像
conda config --show channels
  1. 添加镜像(这里使用清华的)
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge 
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/msys2/
  1. 恢复默认源
conda config --remove-key channels

接下来将conda激活到要安装的环境,我的是d2l

conda activate d2l

安装cuda

查找源上可用的cuda版本

conda search cudatoolkit --info

找到想要下载的cuda版本后,把cuda下载到本地:复制url字段里的下载链接到浏览器,会自动下载到本地Download文件夹下。
然后安装本地包:

conda install --use-local 本地cuda包所在的路径

安装cudnn

查看cuda对应的cudnn版本

conda search cudnn --info


我的cuda版本是11.3,所以选择这个。
同样,将url复制到浏览器下载,然后conda本地安装

conda install --use-local 本地cudnn包所在的路径

安装pytorch

去官网查询与cuda匹配的pytorch和torchvision包。官网:INSTALLING PREVIOUS VERSIONS OF PYTORCH
我的cuda是11.3,因此我执行这句话:

pip install torch==1.12.1+cu113 torchvision==0.13.1+cu113 torchaudio==0.12.1 --extra-index-url https://download.pytorch.org/whl/cu113

注意,这里不要用conda的命令,而是用pip的命令原因是conda会让你下载cpu版本的torch[1]
等一会儿就安装成功了。

验证

输入python代码

(d2l) PS E:\study\deapLearning\d2l-zh> python
Python 3.8.13 (default, Oct 19 2022, 22:38:03) [MSC v.1916 64 bit (AMD64)] :: Anaconda, Inc. on win32
Type "help", "copyright", "credits" or "license" for more information.
>>> import torch
>>> torch.__version__
'1.12.1+cu113'
>>> torch.cuda.is_available()
True
>>> torch.cuda.device_count()
1
>>>

成功。


参考了以下博文,在此表示感谢:
conda虚拟环境中安装cuda和cudnn,再也不用头疼版本号的问题了
Anaconda创建虚拟环境报错—UnavailableInvalidChannel: The channel is not accessible or is invalid
torch.cuda.is_available()返回false——解决办法


  1. 来源是torch.cuda.is_available()返回false——解决办法 ↩︎

标签:--,torch,python,conda,cuda,安装
From: https://www.cnblogs.com/white514/p/16872685.html

相关文章

  • 2 zookeeper安装
    zookeeper包下载:https://mirror.bit.edu.cn/apache/zookeeper/1下载解压,修改配置zookeeper下载:http://archive.apache.org/dist/zookeeper/https://www.aboutyun.co......
  • 在线安装redis
    redis允许外网访问设置1.取消绑定本地地址找到redis配置文件,redis.conf,注释掉指定的bind,当不指定时表示允许所有访问。2.关闭保护模式在redis服务器上使用redis-cli,执行命......
  • mysql rpm安装
    目录mysqlrpm安装安装包下载安装mysqlmysqlrpm安装安装包下载官网:https://dev.mysql.com/downloads/mysql/下载完成后你应该拥有下面四个包文件安装mysql//安装......
  • Docker安装Mysql8
    ​​CentOS7.9安装Docker​​拉取mysql镜像搜寻仓库里面有那些镜像dockersearchmysql我们拉取mysql:latest的镜像dockerpullmysql:latest查看镜像但是我们不知道m......
  • 4.jenkins安装配置升级
    1.什么是jenkinsjenkins是一个开源提供友好操作界面的持续集成的工具,由java开发而成。jenkins是一个调度平台,本身不处理任何事情,都是借由插件完成所有的工作2.为什么使......
  • Docker 安装 Jenkins
    一、Docker介绍1、什么是dockerDocker容器是一个开源的应用容器引擎,让开发者可以以统一的方式打包他们的应用以及依赖包到一个可移植的容器中,然后发布到任何安装了docke......
  • windows版数据库mysql的安装
    一、下载MySQLMysql官网下载地址:MySQL::DownloadMySQLInstaller(ArchivedVersions)1.选择要安装的版本,本篇文章选择的是5.7.31版本,点击Download下载 正在上传......
  • vue之cli脚手架安装和webpack-simple模板项目生成
    ue-cli是一个官方发布vue.js项目脚手架,使用vue-cli可以快速创建vue项目。GitHub地址是:​​https://github.com/vuejs/vue-cli​​一、安装Nodejs首先需要安装Node环......
  • Ubuntu离线安装 MySQL 5.7
    来源:https://www.ngui.cc/article/show-581968.html?action=onClick一、所需环境操作系统:Ubuntu 20.04数据库:MySQL5.7.34网络情况:内网服务器,无法访问互联网资源......
  • RocketMQ学习-服务安装、及服务端命令学习
    写在前面简单说明一下,RocketMQ是阿里团队开发,并推到Apache社区,它的入门稍微有点难(主要是环境、网络、社区的资源有限),下面我简单介绍一下一、安装、部署这里包括三个Windows......