windows10下cuda10.0+pytorch安装

windows10下cuda10.0+pytorch安装

win10查看cuda当前版本

  • 控制面板找到NVDIA控制面板
  • 窗口左下角系统信息
  • 选择组件选项,查看文件名为NVCUDA.DLL的产品名称CUDA后面跟的就是版本。(我8.0.0)
    windows10下cuda10.0+pytorch安装_第1张图片

下载CUDA10.0

(查看CUDA Toolkit历史版本官方链接:https://developer.nvidia.com/cuda-toolkit-archive )

选择系统对应的版本 https://developer.nvidia.com/cuda-10.0-download-archive?target_os=Windows&target_arch=x86_64&target_version=10&target_type=exelocal

环境变量设置

环境变量-系统变量中

新加了CUDA_PATH和CUDA_PATHV10_0

再添加:

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

输入 nvcc -V 显示驱动版本信息。

安装指定版本pytorch

看pytorch官网: https://pytorch.org/get-started/locally

哦豁发现官网中当前支持的cuda版本10.1,千辛万苦下的10.0,担心版本不匹配直接在线可能下完也GG。

来到网页 https://download.pytorch.org/whl/torch_stable.html

找寻自己适合的版本下载至本地:我选择 cu100/torch-1.0.1-cp36-cp36m-win_amd64.whl

(显然:cu后面是cuda版本10.0/torch版本-python版本-电脑系统版本)

【emmmmmm但是这个下载时间还是太长了点,吃完饭睡个觉醒过来 就下完了】

pip直接安装:

cmd到下载目录下:pip install XXX.wheel

安装后python环境里输入:import torch查看

但是因为安装完import pytorch报错DLL,网上搜索发现要更换python版本。emmmm然而我电脑只安装了一个python版本我不想删,然后躲不开的又来到了anaconda。

anaconda安装:

anaconda安装参考 https://blog.csdn.net/ITLearnHall/article/details/81708148

下载地址使用清华源,有梯子也不建议用官网慢哭了:https://mirrors.tuna.tsinghua.edu.cn/anaconda/archive/?C=M&O=D

HTTPError坑解决见: https://blog.csdn.net/Insylol/article/details/88798061

一些anaconda常见命令:
#会列出当前安装的所有pyhon环境
conda info -e   
#创建并激活一个环境
conda create --name [environ name] python=[版本号]  
# 激活环境
activate [environ name] 
# 查看当前python版本
python -V  
# 退出
deactivate 

进入anaconda py3.6.X的环境里:再次cd到pytorch的.whl安装目录,pip install。根据提示升级相关内容。

最后测试成功
在这里插入图片描述

你可能感兴趣的:(windows10下cuda10.0+pytorch安装)