Win10+Anaconda3+python3.6+cuda10.1+cudnn7.6.0+pytorch1.7.0+torchvision0.8.0安装

Win10+Anaconda3+python3.6+cuda10.1+cudnn7.6.0+pytorch1.7.0+torchvision0.8.0安装

  • 一、下载Anaconda
  • 二、安装python3.6
  • 三、安装Cuda
  • 四、安装cudnn
  • 五、安装pytorch
  • 六、报错处理
    • 1. 添加cuda环境变量报错
    • 2. torch和torchvision出现False
    • 3.torch.cuda.is_available()在Anaconda Prompt中为True,在pycharm中为False
    • 4.pycocotools报错

一、下载Anaconda

1.打开Anaconda清华镜像网站:https://mirrors.tuna.tsinghua.edu.cn/anaconda/archive/
选择合适的Anaconda版本,python3.6对应的是Anaconda3-5.2.0,5.3以后的都是python 3.7的,按需求下载。如下图所示:
Win10+Anaconda3+python3.6+cuda10.1+cudnn7.6.0+pytorch1.7.0+torchvision0.8.0安装_第1张图片
2.下载好后打开exe文件,点击next
Win10+Anaconda3+python3.6+cuda10.1+cudnn7.6.0+pytorch1.7.0+torchvision0.8.0安装_第2张图片
3.点击I agree
Win10+Anaconda3+python3.6+cuda10.1+cudnn7.6.0+pytorch1.7.0+torchvision0.8.0安装_第3张图片
4.选择Just Me
Win10+Anaconda3+python3.6+cuda10.1+cudnn7.6.0+pytorch1.7.0+torchvision0.8.0安装_第4张图片
5.默认安装路径,点击next
Win10+Anaconda3+python3.6+cuda10.1+cudnn7.6.0+pytorch1.7.0+torchvision0.8.0安装_第5张图片
6.勾选第二条,点击Install,等待安装。
Win10+Anaconda3+python3.6+cuda10.1+cudnn7.6.0+pytorch1.7.0+torchvision0.8.0安装_第6张图片

二、安装python3.6

1.下载好Anaconda后,打开Anaconda prompt,输入如下语句,进行安装:

conda create -n pytorch python=3.6  # pytorch为虚拟环境名,可任意取

安装过程中出现如下选择,都选择y
Win10+Anaconda3+python3.6+cuda10.1+cudnn7.6.0+pytorch1.7.0+torchvision0.8.0安装_第7张图片

2.验证是否安装成功:输入语句

conda activate pytorch

出现如下界面,即表示安装成功。
Win10+Anaconda3+python3.6+cuda10.1+cudnn7.6.0+pytorch1.7.0+torchvision0.8.0安装_第8张图片

三、安装Cuda

1.首先查看电脑的英伟达显卡支持的cuda版本,具体步骤如下:
①打开控制面板,点击硬件和声音Win10+Anaconda3+python3.6+cuda10.1+cudnn7.6.0+pytorch1.7.0+torchvision0.8.0安装_第9张图片
②点击NVIDIA控制面板
Win10+Anaconda3+python3.6+cuda10.1+cudnn7.6.0+pytorch1.7.0+torchvision0.8.0安装_第10张图片
③点击上方的帮助,再点击系统信息。
Win10+Anaconda3+python3.6+cuda10.1+cudnn7.6.0+pytorch1.7.0+torchvision0.8.0安装_第11张图片
④点击组件,可查看支持的cuda版本
Win10+Anaconda3+python3.6+cuda10.1+cudnn7.6.0+pytorch1.7.0+torchvision0.8.0安装_第12张图片
2.下载cuda:https://developer.nvidia.com/cuda-toolkit-archive
选择显卡对应版本的cuda,请确保C盘有足够的空间
Win10+Anaconda3+python3.6+cuda10.1+cudnn7.6.0+pytorch1.7.0+torchvision0.8.0安装_第13张图片
3.选择相应的配置进行下载。Win10+Anaconda3+python3.6+cuda10.1+cudnn7.6.0+pytorch1.7.0+torchvision0.8.0安装_第14张图片
4.下载好后,双击exe文件
①默认路径
Win10+Anaconda3+python3.6+cuda10.1+cudnn7.6.0+pytorch1.7.0+torchvision0.8.0安装_第15张图片
Win10+Anaconda3+python3.6+cuda10.1+cudnn7.6.0+pytorch1.7.0+torchvision0.8.0安装_第16张图片
②点击同意并继续
Win10+Anaconda3+python3.6+cuda10.1+cudnn7.6.0+pytorch1.7.0+torchvision0.8.0安装_第17张图片
③选择精简
Win10+Anaconda3+python3.6+cuda10.1+cudnn7.6.0+pytorch1.7.0+torchvision0.8.0安装_第18张图片
④勾选I understand…
Win10+Anaconda3+python3.6+cuda10.1+cudnn7.6.0+pytorch1.7.0+torchvision0.8.0安装_第19张图片
⑤等待安装完成
Win10+Anaconda3+python3.6+cuda10.1+cudnn7.6.0+pytorch1.7.0+torchvision0.8.0安装_第20张图片

⑥测试安装成功:
cmd输入:

nvcc -V

出现如下界面,即安装成功。
Win10+Anaconda3+python3.6+cuda10.1+cudnn7.6.0+pytorch1.7.0+torchvision0.8.0安装_第21张图片
5.添加环境变量(参考:https://zhuanlan.zhihu.com/p/158900161)
打开电脑->属性->高级系统设置->环境变量:
①将下面的内容新建到系统变量之中:

CUDA_SDK_PATH = C:\ProgramData\NVIDIA Corporation\CUDA Samples\v10.1
CUDA_LIB_PATH = %CUDA_PATH%\lib\x64
CUDA_BIN_PATH = %CUDA_PATH%\bin
CUDA_SDK_BIN_PATH = %CUDA_SDK_PATH%\bin\win64
CUDA_SDK_LIB_PATH =%CUDA_SDK_PATH%\common\lib\x64

Win10+Anaconda3+python3.6+cuda10.1+cudnn7.6.0+pytorch1.7.0+torchvision0.8.0安装_第22张图片
②然后再找到系统变量中的path,选择编辑。添加以下内容:

%CUDA_LIB_PATH%;%CUDA_BIN_PATH%;%CUDA_SDK_LIB_PATH%;%CUDA_SDK_BIN_PATH%
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1\lib\x64
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1\bin
C:\ProgramData\NVIDIA Corporation\CUDA Samples\v10.1\common\lib\x64
C:\ProgramData\NVIDIA Corporation\CUDA Samples\v10.1\bin\win64;

四、安装cudnn

1.下载地址:https://developer.nvidia.com/rdp/cudnn-download
2.注册或者登录NVIDIA账号,注册需要填写一个简单的问卷。
3.选择cuda对应版本的cudnn进行下载
Win10+Anaconda3+python3.6+cuda10.1+cudnn7.6.0+pytorch1.7.0+torchvision0.8.0安装_第23张图片
4.下载好后,解压压缩包,将里面的include、lib、bin三个文件分别粘贴到cuda安装目录下的相应文件内。
Win10+Anaconda3+python3.6+cuda10.1+cudnn7.6.0+pytorch1.7.0+torchvision0.8.0安装_第24张图片
5.测试是否安装成功cudnn
启动cmd或者powershell,cd到安装目录下的 …\extras\demo_suite,然后分别执行bandwidthTest.exe和deviceQuery.exe,应该得到下图(参考:https://blog.csdn.net/fengxinzioo/article/details/105646969):
Win10+Anaconda3+python3.6+cuda10.1+cudnn7.6.0+pytorch1.7.0+torchvision0.8.0安装_第25张图片

五、安装pytorch

本来用的是conda自带的源,但是由于下载太慢,换成了清华源,结果https报错,好像不能用。网上找了很多方法,终于成了:先将pytorch和torchvision的安装文件下载好,然后后离线安装
1.下载网站:https://download.pytorch.org/whl/torch_stable.html
根据python版本和cuda版本选择相应的torch和torchvision的wml文件下载,版本对应可以查看pytorch官网:https://pytorch.org/get-started/previous-versions/。选择的时候要看下前面是否带“cu”,版本不要下载错
Win10+Anaconda3+python3.6+cuda10.1+cudnn7.6.0+pytorch1.7.0+torchvision0.8.0安装_第26张图片
Win10+Anaconda3+python3.6+cuda10.1+cudnn7.6.0+pytorch1.7.0+torchvision0.8.0安装_第27张图片
2.打开Anconda Prompt,输入pip install {.whl文件路径}安装torch,然后安装torchvision。注意:先装torch再装torchvision
3.验证是否安装成功
打开Anconda Prompt,输入如下代码:

import torch
import torchvision
torch.cuda.is_available()

若输出True如下所所示,则表示安装成功!
Win10+Anaconda3+python3.6+cuda10.1+cudnn7.6.0+pytorch1.7.0+torchvision0.8.0安装_第28张图片

六、报错处理

1. 添加cuda环境变量报错

如果遇到如下错误:此环境变量太大。此对话框允许将值设置为最长 2047 个字符,可参考这篇文章:https://zhuanlan.zhihu.com/p/364186011

2. torch和torchvision出现False

验证torch和torchvision的时候出现False,报错内容如下:

 UserWarning: CUDA initialization: CUDA unknown error - this may be due to an incorrectly set up environment, e.g. changing env variable CUDA_VISIBLE_DEVICES after program start. Setting the available devices to be zero. (Triggered internally at  /opt/conda/conda-bld/pytorch_1614378083779/work/c10/cuda/CUDAFunctions.cpp:109.)
  return torch._C._cuda_getDeviceCount() > 0

原因
nvidia-modprobe 版本过低或未安裝,与显卡驱动的版本不匹配。
解决方法
1.以Win10为例,首先我们鼠标右键点击“此电脑”,选择“管理”,在计算机管理中,我们点击“设备管理器”,展开“显示适配器”,找到并右键您的显卡设备,选择“更新驱动程序”(也可以选择属性–驱动程序–更新驱动程序),如下图所示。
在这里插入图片描述
2.选择第一个“自动搜索驱动程序”
在这里插入图片描述
3.等待片刻,就可以更新显卡驱动程序,会提示正在联机搜索驱动程序,更新好了之后,会提示“你的设备的最佳驱动程序已安装”。
4.更新好之后,再次验证pytorch,出现True即表示安装成功。

3.torch.cuda.is_available()在Anaconda Prompt中为True,在pycharm中为False

原因
Pycharm环境配置错误。
解决方法
设置Pycharm编辑器,选择base环境(和Anaconda Prompt运行时一样的环境)。

4.pycocotools报错

① ModuleNotFoundError: No module named ‘pycocotools._mask‘ 错误

pip install pycocotools-windows

你可能感兴趣的:(python,pytorch,windows)