win10python安装步骤_Win10 Python安装过程,win10pytorch

查看一些重要参数

参数名称

查看方法

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

控制面板->卸载程序->所有带NVIDIA的全部卸载(有些可能会要求卸载完立即重启,就多重启几次,不要怕耽误时间)

卸载完后在

C:\Program Files

C:\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

至此安装就结束了~~

你可能感兴趣的:(win10python安装步骤)