windows下配置pytorch环境(CUDA10.1)


参考文章
CUDA、CUDNN在windows下的安装及配置
如何搭建Pytorch环境(Windows版)
windows conda 安装 PyTorch 切换清华源镜像教程
windows系统下pytorch安装教程

一、安装CUDA前的准备

(1)查看自己N卡支持的CUDA版本,打开NVIDIA控制面板,选择系统信息

在这里插入图片描述

NVIDIA控制面板下 系统信息 显示:CUDA11.7。这里CUDA11.7是支持的最高版本的CUDA,可以向下兼容,且可以安装多个版本的CUDA,可以通过更改环境变量来更改为你需要用到的CUDA版本。CUDA多版本共存请参考博客: Windows下CUDA多版本共存

如果版本过低则需要升级驱动。NVIDIA驱动下载
在这里插入图片描述
例如我的显卡是NVIDIA GeForce 930M(Windows10)然后点击search。
在这里插入图片描述
download。
在这里插入图片描述
download。
下载完成后,找到执行文件,双击进行安装。(这里就不再赘述,按步骤安装即可。)
在这里插入图片描述
安装完成后,再去NVIDIA控制面板检查自己的CUDA版本,现在就是最新版本了。
在这里插入图片描述

(2)NVIDIA官网下载对应CUDA版本

由于我的电脑已经用了七八年,担心不能支持最新的CUDA,所以选择了一个较老版本CUDA10.1。
CUDA下载地址:CUDA Toolkit Archive | NVIDIA Developer
在这里插入图片描述
在这里插入图片描述

(3)CUDNN的下载

官网下载地址:cuDNN Download
没有账号需要注册账号,账号登录后,找到自己对应的CUDA版本,点击下载。
在这里插入图片描述
在这里查看旧版本。
在这里插入图片描述
选了个CUDA10.1的比较旧的,应该也可以选对应版本的新的但是没试过。

二、安装过程

这部分的内容看CUDA、CUDNN在windows下的安装及配置的第二部分就好,作者写的很详细。

三、安装Anaconda

Anaconda清华源镜像下载地址
在这里插入图片描述

切换清华源镜像

conda config --add channels http://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/win-64
conda config --add channels http://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/win-64
conda config --set show_channel_urls yes
conda config --add channels http://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/win-64

四、安装pytorch

pytorch旧版本在线下载方式

pytorch安装成功验证

python
import torch
import torchvision

  • 0
    点赞
  • 6
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值