PyTorch&CUDA安装过程及测试

2023-11-09

PyTorch&CUDA安装过程及测试

1 准备工作

1.1 Anaconda&NVDIA驱动

之前安装过了Anaconda和NVIDIA驱动,所以直接跳过这几步。

打开NVIDIA控制面板,帮助–>系统信息–>组件,查看自己的CUDA版本号。在这里插入图片描述

或者进入cmd,输入

nvidia-smi

也可以查看

在这里插入图片描述

1.2 安装CUDA

注意装CUDA前一定要有VS!!

  1. 基于我们主机的CUDA version,从官网下载CUDA Tookit 11.1

https://developer.nvidia.com/cuda-11.1.1-download-archive

在这里插入图片描述

  1. 进行安装,路径默认

在这里插入图片描述

​ 选择自定义安装

image-20220428223150574

不选Visual Studio Integration

开发的时候vs里面没有cuda模块,原来需要先装vs再装cuda。。。只能重新安装一遍cuda了

image-20220428223247351

安装位置

image-20220428223324706

安装完成

image-20220428224508955

系统变量中会自动添加CUDA_PATH

image-20220428224552176

查看cuda版本

nvcc --version

image-20220428224639595

查看环境变量

set cuda

image-20220428224808643

2 创建PyTorch虚拟环境

打开Anaconda Prompt

image-20220428162959902

创建环境,环境名设置为PyTorch

conda create -n PyTorch python=3.8

image-20220428163418208

成功后,查看环境

conda info --envs

image-20220428163546872

3 安装PyTorch

3.1 conda换源

一开始没有换源,速度奇慢

Windows 用户无法直接创建名为 .condarc 的文件,可先执行 conda config --set show_channel_urls yes 生成该文件之后再修改。

生成.condarc文件

conda config --set show_channel_urls yes

image-20220428193449569

之后可以在C:\Users\xxx中看到.condarc文件,记事本打开 .condarc 文件,重写其中的内容:

channels:
  - defaults
show_channel_urls: true
default_channels:
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/r
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/msys2
custom_channels:
  conda-forge: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
  msys2: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
  bioconda: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
  menpo: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
  pytorch: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
  simpleitk: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud

image-20220428193431317

运行 conda clean -i 清除索引缓存,保证用的是镜像站提供的索引:

conda clean -i

image-20220428193624419

3.2 conda安装PyTorch

Anaconda进入PyTorch环境:

conda activate PyTorch

image-20220428164121099

conda install

conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch

image-20220428164304280

4 测试环境

4.1 测试torch.cuda.is_available

打开 Anaconda prompt 命令窗口,激活环境,输入python,进入python开发环境中。

# 测试
import torch
torch.cuda.is_available()
坑1:未安装cuda

第一次因为没有安装cuda,所以False,需要步骤1.2中的CUDA安装过程:

image-20220428200458636

坑2:cuda和PyTorch版本不对

安装了cuda还是不行,看一下是不是版本不对

image-20220428231123649

尝试替换torch版本,我的cuda版本是11.1

image-20220428231638005

image-20220428231607769

image-20220428232629501

重装之后终于成功~

image-20220429001239266

4.2 cublas库使用测试

编写cublastest.cu程序,并编译,注意使用cublas库时,编译命令中要加上-lcublas

nvcc -lcublas cublastest.cu -o cublastest

image-20220505142357460
运行编译生成的可执行文件

./cublastest

image-20220505142556550

参考

Conda安装过程中InvalidArchiveError

PyTorch 最新安装教程(2021-07-27)

安装torch后,torch.cuda.is_available()结果为false的问题

本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系:hwhale#tublm.com(使用前将#替换为@)

PyTorch&CUDA安装过程及测试 的相关文章

随机推荐

  • ELK —— Logstash 将 MySQL 数据同步至 ElasticSearch

    文章目录 一 搭建环境 1 0 环境依赖 1 1 搭建 ElasticSearch 环境 1 1 1 ElasticSearch 简介 1 1 2 启动 ElasticSearch 1 2 Logstash 多数据源采集系统 1 3 Kib
  • 【C++ Primer Plus学习记录】共用体

    共用体 union 是一种数据格式 它能够存储不同的数据类型 但只能同时存储其中的一种类型 也就是说 结构可以同时存储int long和double 共用体只能存储int long或double 共用体的句法与结构相似 但是含义不同 例如
  • 可有可无的Mysql工作技巧 2

    你有一个订单表 CREATE TABLE sales engineering order tmp id VARCHAR 32 NOT NULL COMMENT 主键ID order no VARCHAR 50 NOT NULL COMMEN
  • Unity3D为3D物体添加点击事件

    1 相机添加脚本 ClickEvent using System Collections using System Collections Generic using UnityEngine using UnityEngine EventS
  • sqlserver数据库迁移到mysql的方法和步骤

    一 迁移方法 工具 用mysql yog 下载地址1 https download csdn net download LT 1029 12261398 下载地址2 http yunpan cn QazFaTkuUpwmf 提取码 2310
  • 编写一个函数,用来删除字符串中的空格

    程序代码 include
  • 苏小红版 c语言程序设计(第三版)系列实验题:学生成绩管理系统V2.0

    github https github com Jackie0Feng SAMS 系统需求描述 某班有最多不超过30人 具体人数由键盘输入 参加某门课的考试 用一维数组和函数指针作为函数参数编程实现如下学生成绩管理 1 录入每个学生的学号和
  • 数据库之桥

    前言 通俗地讲 就是如何使用 工具 eclipse 连接到 你的数据库 JDBC 简介 全称为 Java Database Connectivity 可以使用Java语言去操作数据库 通过数据库提供的Connector作为支持 对于不同的语
  • MyBatis中自动生成主键

    MyBatis中自动生成主键 详细 Mybatis 配置文件配置useGeneratedKeys 参数只针对 insert 语句生效 默认为 false 在INSERT语句中 我们为可以自动生成 auto generated 主键的列 ST
  • Maven镜像(mirror)

    使用镜像代替中央仓库 国内开发人员由于网络原因 直接从中央仓库下载构件时 速度较慢或不稳定 我们通常会使用中央仓库的国内镜像站来解决该问题 配置 Maven 镜像的方法也非常的简单 我们只需要在 Maven 安装目录中 setting xm
  • 如何更改git提交代码的用户名称

    桌面单击鼠标右键 点击Git Bash Here 出现弹窗 修改git提交者的名称 git config global user name 要修改的名称 查看git提交者的名称 git config user name
  • Python: 生成带用户昵称的头像

    Python 生成带用户昵称的头像 需求 新建用户后 根据用户输入的昵称生成图片 例 注册 钉钉 用户后 头像根据输入的名字生成 开发环境 Windows 10 Python 3 8 Pillow 8 1 2 实现 蛇皮皮蛋 Python创
  • 【千律】C++基础:函数的重载

    include
  • 易游网吧无盘服务器安装,易游无盘安装教程

    易游无盘安装教程 2021 02 19 00 59 49 简介 php去除nbsp的方法 首先创建一个PHP代码示例文件 然后通过 preg replace s nbsp xc2 xa0 strip tags val 方法去除所有nbsp即
  • Sublime Text3 SublimeREPL python

    首先什么是REPL A Read Eval Print Loop REPL is available both as a standalone program and easily includable in other programs
  • qt之利用lineEdit设置密码样式

    参考博客 https blog csdn net u012288722 article details 77850974 主要参考以上大神博客实现如下 可能需要加上头文件 include
  • v8学习---添加js全局函数

    v8学习 添加js全局函数 标签 v8 2013 11 10 10 11 951人阅读 评论 0 收藏 举报 分类 v8 12 版权声明 本文为博主原创文章 未经博主允许不得转载 cpp view plain copy include
  • URL 链接中 井号#、问号?、连接符& 分别有什么作用?

    井号 表示网页中的一个位置 被称之为锚点 常用于某个网页间不同位置的跳转 简单的说就是在一个网页中 URL 不变的情况下 通过添加 buy 的字符在 URL 最后可以跳转到当前网页中已经定义好的锚点 id buy 位置 同样 的改变也会增加
  • APPIcon生成

    有两种生成方法 比较常用 一种是设计师常用的 在线生成图标 https icon wuruihong com 支持 jpg png psd 文件 可以生成圆角 可以自定义大小 生成后可以预览 模拟手机屏幕的效果预览 第二种就是终端生成 利用
  • PyTorch&CUDA安装过程及测试

    PyTorch CUDA安装过程及测试 1 准备工作 1 1 Anaconda NVDIA驱动 之前安装过了Anaconda和NVIDIA驱动 所以直接跳过这几步 打开NVIDIA控制面板 帮助 gt 系统信息 gt 组件 查看自己的CUD