win10 pytorch安装过程

查看一些重要参数

参数名称查看方法
CUDA版本查看如果已经安装过CUDA可以通过命令nvcc -V 可以查看当前CUDA的版本
cuDNN版本查看已经安装过cuDNN可以在默认安装位置C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.1\inculde\bin\cudnn.h 用记事本打开,可以看到cuDNN的版本(好像旧版本的可以,新版本的看不到了)
显卡型号查看我的电脑->属性->设备管理器->显示适配器查看在这里插入图片描述
查看NVIDIA驱动对应的CUDA版本右击鼠标NVIDIA控制面板->左下角系统信息->组件->可以查看对应CUDA版本在这里插入图片描述

坎坷经历

曾经安装过NVIDIA驱动,可能因为windows自动更新导致不可用,一开始以为NVIDIA、CUDA、cuDNN这些都有,就直接去安装pytorch了……结果我的NVIDIA控制面板都打不开
所以先检查一下自己的NVIDIA可不可用!!

安装NVIDIA驱动

可以卸载掉所有的NVIDIA驱动重新安装

卸载NVIDIA

  1. 控制面板->卸载程序->所有带NVIDIA的全部卸载(有些可能会要求卸载完立即重启,就多重启几次,不要怕耽误时间)
  2. 卸载完后在C:\Program FilesC:\Program Files(x86)中所有带NVIDIA的文件夹全部删掉

安装NVIDIA驱动

先查看自己的显卡型号,然后到NVIDIA官网输入自己的显卡型号,搜索对应的NVIDIA驱动下载(我就直接下载了第一个),下载完傻瓜式安装就可以,建议不要更改默认路径
在这里插入图片描述
在这里插入图片描述

安装CUDA

查找NVIDIA对应的CUDA版本,CUDA官网下载,这里是11.1版本的,之前的版本可能要自己找一下,下载完后同样傻瓜式安装

安装cuDNN

同样官网下载CUDA版本对应的cuDNN
注意看清对应的CUDA版本再下载
下载完后是一个压缩包,解压之后,分别将压缩包内的bin、include、lib\x64中的文件,复制粘贴到CUDA路径(C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.1)下对应的同样名称的文件夹里即可

安装pytorch

首先使用anaconda建一个python36的虚拟环境 conda create -n pytorch python=3.6
然后 activate pytorch
pytorch官网中选择你操作系统、CUDA版本等、最下方会出来安装命令
conda install pytorch torchvision torchaudio cudatoolkit=11.0 -c pytorch
在这里插入图片描述

复制然后切换回cmd粘贴命令安装即可

检查是否成功安装

import torch不报错说明可以在cpu下使用pytorch
torch.cuda.is_availabel()返回True说明可以在GPU下使用pytorch

在这里插入图片描述

至此安装就结束了~~

  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 1
    评论
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值