admin管理员组文章数量:1562635
1、安装环境
OS: Win10 专业 x64
Python: Python 3.7.7(通过Anaconda软件自带安装)
Anaconda: Anaconda3-2020.02-Windows-x86_64.exe 【这个版本自带的python是3.7.7】
CUDA: cuda_10.1.105_418.96_windows.exe ,根据操作系统不同,可以在线选择不同版本下载,界面网址为下载地址 【我原先用的是10.2,后面有图片不一致的请原谅】
Pytorch:官方是Pytorch在线指令安装,建议参考我后面的离线手动安装,速度比较快。
Pycharm:pycharm-community-2021.1.3.exe,开发工具,社区免费版,下载后,自行一路默认安装。
2、Anaconda安装过程
1.选择安装路径:C:\ProgramData\Anaconda3
2.勾选两个:选择添加环境变量,设置Anaconda3自带的Python3.7的为系统默认环境。
3.打开windows自带的命令窗口cmd.exe,测试安装是否成功
cmd窗口中输入命令:conda list,输出结果如下:
cmd窗口中输入命令:conda --version,结果输出版本号,代表安装成功。
3、CUDA安装过程
1.CUDA下载
百度搜索CUDA download,进入官方下载页面,选择适合的操作系统和install type:(local的exe),具体网址为下载地址。
2、双击安装
检查通过的话,选中需要的Visual Studio 环境,进入安装环节。(万一安装失败,请卸载干净后,重启系统,重新安装。)
3.检查cuda是否安装成功
先核实一下,目录C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1\bin下,是否有nvcc.exe文件。
然后在cmd中输入指令: nvcc -V,可以看到我们版本号为Cuda 10.1
本质上,是我们系统环境变量里多了Cuda的路径配置和Anaconda的路径
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1\bin;C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1\libnvvp;
C:\ProgramData\Anaconda3;C:\ProgramData\Anaconda3\Library\mingw-w64\bin;C:\ProgramData\Anaconda3\Library\usr\bin;C:\ProgramData\Anaconda3\Library\bin;C:\ProgramData\Anaconda3\Scripts;
4安装Pytorch
1.首先进入Pytorch下载页面:Start Locally | PyTorch
选择合适的版本10.1,然后拷贝命令:
conda install pytorch torchvision torchaudio cudatoolkit=10.1 -c pytorch
更新conda版本
Please update conda by running
$ conda update -n base -c defaults conda
2、手动下载安装包
下载地址:
Index of /anaconda/pkgs/main/win-64/ | 清华大学开源软件镜像站 | Tsinghua Open Source Mirror(稳定,只需要下载conda文件)
Index of /anaconda/cloud/pytorch/win-64/ | 清华大学开源软件镜像站 | Tsinghua Open Source Mirror (最新,只能下载tar.bz2文件)
https://repo.anaconda/pkgs/main/win-64/(官网)
具体下载安装包列表:
cudatoolkit pkgs/main/win-64::cudatoolkit-10.1.243-h74a9793_0 下载:cudatoolkit-10.1.243-h74a9793_0.tar.bz2
libuv pkgs/main/win-64::libuv-1.40.0-he774522_0 速度快,可以自动安装
ninja pkgs/main/win-64::ninja-1.10.1-py37h7ef1ec2_0 速度快,可以自动安装
pytorch pytorch/win-64::pytorch-1.7.0-py3.7_cuda101_cudnn7_0 下载:pytorch-1.7.0-py3.7_cuda101_cudnn7_0.tar.bz2
torchaudio pytorch/win-64::torchaudio-0.7.0-py37 下载:torchaudio-0.7.0-py37.tar.bz2
torchvision pytorch/win-64::torchvision-0.8.1-py37_cu101 下载:torchvision-0.8.1-py37_cu101.tar.bz2
typing_extensions pkgs/main/noarch::typing_extensions-3.7.4.3-py_0 速度快,可以自动安装
3.拷贝至指定路径下:C:\ProgramData\Anaconda3\pkgs
4.然后手动离线,按顺序安装下载包。
以管理身份打开cmd,输入cmd C:\ProgramData\Anaconda3\pkgs,进入安装包目录。然后按照顺序安装下载包
conda install --use-local pytorch-1.7.0-py3.7_cuda101_cudnn7_0.tar.bz2
conda install --use-local torchvision-0.8.1-py37_cu101.tar.bz2
conda install --use-local torchaudio-0.7.0-py37.tar.bz2
conda install --use-local cudatoolkit-10.1.243-h74a9793_0.tar.bz2
安装成功结果:
5. 执行pytorch安装总指令,检查一遍是否完全安装
执行pytorch安装总指令,检查一遍是否完全安装。
conda install pytorch torchvision torchaudio cudatoolkit=10.1 -c pytorch (红色-c pytorch 去掉,免得从镜像上重新下载)
conda install pytorch torchvision torchaudio cudatoolkit=10.1
6. 验证pytorch是否安装成功
在开始菜单中,打开 Python控制台。
(1)输入 import torch
,如果没有报错,意味着 PyTorch 已经顺利安装了(我的笔记本win10 home版本系统缺少VC_redist.64.exe,安装后也解决)
(2)接下来,输入 torch.cuda.is_available()
,如果是 True,意味着你可以使用 GPU,如果是 False,意味着只能使用CPU。
(换了幅图,原图代码torch.cuda.is_available(),忘加括号了)
本文标签: Pytorch
版权声明:本文标题:Win10 + CUDA10.1+pytorch手动安装 内容由热心网友自发贡献,该文观点仅代表作者本人, 转载请联系作者并注明出处:https://m.elefans.com/dianzi/1727484998a1117111.html, 本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌抄袭侵权/违法违规的内容,一经查实,本站将立刻删除。
发表评论