Pytorch安装及环境配置详细教程(CUDA版本)

news/2024/10/31 5:29:09/

文章目录

  • 前言
  • 一、查看GPU支持的CUDA版本
  • 二、安装CUDA
  • 三、确定torch、torchvision和python版本
  • 四、安装anaconda
  • 五、安装torch和torchvision


前言

安装cuda版本的pytorch时踩了不少坑,网上安装pytorch的版本很多,一般的教程都是到pytorch的官网,利用网址和镜像源去安装,问题非常多。也有教离线安装的,但是没有正确的安装顺序,torch版本和torchvision版本对不上,python版本和自己电脑cuda所支持的pytorch版本对不上等等问题。以下是我根据踩过的坑,记录的一个安装教程。


一、查看GPU支持的CUDA版本

控制面板-NVDIA控制面板,单击NVDIA控制面板,有可能会出现如下提示:
在这里插入图片描述
这是因为NVIDIA Display Container LS 服务没打开。打开任务管理器,将其开启。
在这里插入图片描述
之后在单击NVDIA控制面板,在打开的面板中找到系统信息,单击打开
在这里插入图片描述
在这里插入图片描述
便可查看GPU最高支持的CUDA版本.,我的笔记本最高支持的版本为10.0

二、安装CUDA

可以参考该链接:https://blog.csdn.net/Mind_programmonkey/article/details/99688839
非常详细。


三、确定torch、torchvision和python版本

到这里是非常重要的一步 ,其他教程到这里可能就开始安装anaconda,别急,否则容易踩坑。
我们需要先根据安装的CUDA版本,找到对应的torch和torchvision版本,在这个过程中就确定了python的版本。
首先在下载链接:https://download.pytorch.org/whl/torch_stable.html 中找到我们所能安装的torch版本。

在这里插入图片描述
其中,cu100就是cuda10.0,cp37就是python版本3.7。接着在根据以下表格找到对应的torchvision版本。
https://github.com/pytorch/vision#installation

在这里插入图片描述
在这里插入图片描述
torch1.2版本对应的torchvision版本为0.4.0。这时顺便准备好两个whl文件。
在这里插入图片描述
之所以要选确定好python的版本在安装是因为待会安装anaconda时,会涉及到python版本的问题,假如先盲目安装了任意版本的python,再来下载torch和torchvision,就会遇到自己CUDA版本所支持的torch和torchvision的python版本和安装的python版本不匹配。

四、安装anaconda

下载地址:https://mirrors.tuna.tsinghua.edu.cn/anaconda/miniconda

根据上面确定的python版本3.7,选择如下安装文件。
在这里插入图片描述
一路点next就行。
在这里插入图片描述
安装之后在查看conda命令是否可以正常使用
在这里插入图片描述
在这里插入图片描述

五、安装torch和torchvision

首先建一个Python的虚拟环境来安装pytorch,在Anaconda Prompt (miniconda3)命令行输入

conda create -n PyTorchP37 python=3.7

PyTorchP37为自己的命的环境名称,创建好环境以后,输入activate PyTorchP37就进入虚拟环境中。
在这里插入图片描述
接着切换到刚才下载的两个whl文件的安装目录。

pip install torchvision-0.4.0-cp37-cp37m-win_amd64.whl
pip install torch-1.2.0-cp37-cp37m-win_amd64.whl

在这里插入图片描述
在这里插入图片描述
安装完后,验证以下:

pip install torchvision-0.4.0-cp37-cp37m-win_amd64.whl

在这里插入图片描述
打印True, cuda版本Pytorch便安装成功了。


http://www.ppmy.cn/news/21365.html

相关文章

Spring-基础知识二

Spring9.Spring JdbcTemplate的使用9.1 JdbcTemplate入门9.1.1 需要的包9.1.2 代码测试9.2 将数据源和jdbcTemplate交给Spring来管理9.2.1 druid连接池9.2.2 使用外部文件配置数据连接信息9.3 基于JdbcTemplate实现DAO9.Spring的事务管理机制9.1 PlatformTransactionManager 事…

Active Directory计算机备份和恢复

在Active Directory(AD)环境中,用户通过域中的计算机认证他们自身。从AD中删除这些计算机账户时,系统也会自动从域中删除它们。于是,用户不能再通过些计算机登录网络。为允许用户访问域资源,必须恢复这些已…

springboot 集成rabbitmq

1 导入maven依赖jar包 <dependency><groupId>org.springframework.boot</groupId><artifactId>spring-boot-starter-amqp</artifactId> </dependency>2 配置application.properties #mq spring.rabbitmq.host192.168.43.100 spring.rabbit…

这就是传说中超难的N皇后?——详细图解!

✔️本文主题&#xff1a;回溯算法之N皇后 算法 ✔️题目链接&#xff1a;N皇后 详解N皇后一、前言二、题目信息三、解题思路四、参考代码五、结语一、前言 大家好久不见&#xff0c;今天我们一起来学习一道很经典、也很有难度的一道题目——N皇后 二、题目信息 按照国际象棋…

Git提交后代码后修改commit信息

文章目录1. $ git rebase -i HEAD~n2. 执行后显示近n次commit信息3.执行 git commit --amend后会跳出编辑器4.执行$ git rebase --continue修改最近n次提交1. $ git rebase -i HEAD~n 例如&#xff1a;要修改近三次提交&#xff0c;git rebase -i HEAD~3 2. 执行后显示近n次c…

【计网】ip地址和子网掩码

ip和子网掩码两者基本作用ip地址ip地址范围ip地址组成和子网掩码子网掩码为什么需要子网掩码总结两者基本作用 ip地址为接入网络的节点的地址&#xff0c;分为网络号和主机号。 子网掩码则是用来定义/判断ip地址的网络号和主机号。 ip地址 ip地址范围 用我们最常见的ip地址…

Java版数据结构与算法笔记

文章目录一、数据结构与算法概述及题目1、数据结构和算法的关系2、线性结构与非线性结构Ⅰ-线性结构Ⅱ-非线性结构3、经典面试题Ⅰ-字符串匹配问题&#xff1a;Ⅱ-汉诺塔游戏Ⅲ-八皇后问题:Ⅳ-马踏棋盘算法4、几个实际编程中遇到的问题Ⅰ-字符串替换问题Ⅱ-一个五子棋程序Ⅲ-约…

【算法】【位运算模块】使用位运算完成整数的加减乘除

目录前言问题介绍解决方案代码编写java语言版本c语言版本c语言版本思考感悟写在最后前言 当前所有算法都使用测试用例运行过&#xff0c;但是不保证100%的测试用例&#xff0c;如果存在问题务必联系批评指正~ 在此感谢左大神让我对算法有了新的感悟认识&#xff01; 问题介绍 …