(自用)windows安装cuda+pytorch

倒腾好久,出错无数,记录下步骤避免再次踩坑

1.安装anaconda 会自动下一个版本的python

2.conda create --name python=3.6(3.5pip会出错,太高后面好像也会出错)

        创建一个3.6版本的python虚拟环境

        activate虚拟环境

3.找显卡型号如何查看windows的CUDA版本_天泪哈哈的博客-CSDN博客_查看cuda版本

(1)按win+Q,输入控制面板,然后查看方式小图标,点击NVIDIA控制面板:

点击NVIDIA控制面板的帮助,点击左下角系统信息:

点击组件:这里就显示了你的CUDA的信息啦!!!

4.在CUDA Toolkit Archive | NVIDIA Developer找显卡匹配的cuda版本,下载安装

5.更改conda下载源,改为清华的源,否则下载太慢,无限出错

(1)win+r

(自用)windows安装cuda+pytorch_第1张图片

(2)用记事本打开.condarc文件

(3)替换为下面的

channels:
  - http://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
  - http://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge/
  - http://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/msys2/
  - http://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/bioconda/
  - http://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
show_channel_urls: true
ssl_verify: false

6.两种方法下载安装pytorch

(1)pytorch官网找对应的链接win+r     cmd  复制进去下载

 (自用)windows安装cuda+pytorch_第2张图片

(2)先下载whl文件再pip安装(推荐)离线安装:

pytorch官方网站下载torch和torchvision的离线whl安装包,torch和torchvision的版本一定要对应,

https://pytorch.org/get-started/previous-versions/在这里看版本对应

官方安装包下载地址: https://download.pytorch.org/whl/torch_stable.html

下载对应python版本和cuda版本的安装包pytorch和torchvision都下

进入下载文件夹,在文件夹右键 进入命令行,如没有,下方法调出

命令行黑框中输入

activate+虚拟环境名字

然后pip安装

pip install torch-1.9.0+cu111-cp37-cp37m-win_amd64.whl

pip install torchvision-0.10.0+cu111-cp37-cp37m-win_amd64.whl

(自用)windows安装cuda+pytorch_第3张图片

 

按理说就安装好了,输入pip list看看有没有pytorch

输入python

输入  import torch,没有报错即安装成功

输入 torch.cuda.is_available()查看是否能调用gpu

(非必要)调出命令行的步骤:

第一步:新建一个txt文件,命名为OpenCmdHere.txt,注意设置编码格式为ANSI
第二步:在文件中输入如下代码,并保存

Windows Registry Editor Version 5.00

[HKEY_CLASSES_ROOT\Directory\shell\OpenCmdHere]
@="在此处打开命令窗口"
"Icon"="cmd.exe"

[HKEY_CLASSES_ROOT\Directory\shell\OpenCmdHere\command]
@="cmd.exe /s /k pushd \"%V\""

[HKEY_CLASSES_ROOT\Directory\Background\shell\OpenCmdHere]
@="在此处打开命令窗口"
"Icon"="cmd.exe"

[HKEY_CLASSES_ROOT\Directory\Background\shell\OpenCmdHere\command]
@="cmd.exe /s /k pushd \"%V\""

[HKEY_CLASSES_ROOT\Drive\shell\OpenCmdHere]
@="在此处打开命令窗口"
"Icon"="cmd.exe"

[HKEY_CLASSES_ROOT\Drive\shell\OpenCmdHere\command]
@="cmd.exe /s /k pushd \"%V\""

[HKEY_CLASSES_ROOT\LibraryFolder\background\shell\OpenCmdHere]
@="在此处打开命令窗口"
"Icon"="cmd.exe"

[HKEY_CLASSES_ROOT\LibraryFolder\background\shell\OpenCmdHere\command]
@="cmd.exe /s /k pushd \"%V\""

第三步:改后缀为reg  双击运行


 

你可能感兴趣的:(pytorch,windows,深度学习)