Pytorch环境安装【Python3.7+Anaconda3+CUDA10.1】

1、安装环境

OS:              Win7 旗舰版 x64
Python:      Python 3.7.7(通过Anaconda软件自带安装)
Anaconda: Anaconda3-2020.02-Windows-x86_64.exe
CUDA:        cuda_10.1.105_418.96_windows.exe  具体选择界面的下载地址 【我原先用的是10.2,后面有图片不一致的请原谅】

Pytorch:Pytorch安装,建议参考我这篇离线手动安装,速度比较快。Win10 + CUDA10.1+pytorch手动安装》,注意这篇,我后续用的都是win10系统了。

 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下载页面:https://pytorch.org/get-started/locally/

选择合适的版本10.1,然后拷贝命令:

conda install pytorch torchvision cudatoolkit=10.2 -c pytorch

以管理员方式运行2.以管理员身份运行cmd.exe(开始菜单里搜cmd,然后右击以Administrator身份运行)

输入如下脚本命令:

conda install pytorch torchvision cudatoolkit=10.1 -c pytorch

第一次由于网络原因下载失败。。。第二次才好使

3.选择y ,表示同意更新

4. 加速Pytorch的安装(手动添加国内镜像源)

1.打开Anaconda Navigator,点击Channels。

2.手动添加清华镜像源

https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge/
https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/msys2/

一个个将地址add进去,然后点击Update channels。

重新执行步骤4中的pytorch安装脚本,即可提速。(国内实在慢的,建议手动离线安装PKG文件,参考这篇文章《pytorch手动安装开发包》

5. 验证pytorch是否安装成功

在开始菜单中,打开 Python控制台。

(1)输入 import torch,如果没有报错,意味着 PyTorch 已经顺利安装了

(2)接下来,输入 torch.cuda.is_available,如果是 True,意味着你可以使用 GPU,如果是 False,意味着只能使用CPU。

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值