欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

windows10下cuda10.0+pytorch安装

程序员文章站 2022-03-16 18:52:34
...

windows10下cuda10.0+pytorch安装

win10查看cuda当前版本

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

下载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安装