python3.8+pytorch 1.7.0+cuda10.2+cuDNN

本文是在电脑中直接安装cuda10.2的过程,如果是要在anaconda环境中安装的话,可以直接只在虚拟环境中使用命令进行安装,用以下链接中的过程即可。 

在anaconda虚拟环境中安装pytorch和cuda教程


安装过程中看到的不错的教程

全面讲解安装Anaconda+CUDA+PyTorch(含有cuda的具体安装过程)

Windows——安装cuda10.2和cudnn7.6.5(两者搭配稳定可靠,补丁)

win10安装cuda10.2和对应的cudnn版本(cuda和cuDNN完整安装,环境配置简单)

PyTorch和CUDA版本对应关系

aunauhwindows 10 安装cuda11.0 cuDNN(含有环境变量,环境配置比较复杂,参考本文的即可))


1. anaconda3 中 python版本为 3.8

conda create -n env_name python=3.8

2. pytorch 使用conda命令在anaconda中下载

(1)只安装pytorch,没有安装cuda

conda install pytorch==1.7.0 torchvision==0.8.0 torchaudio==0.7.0 -c pytorch

(2)先安装了(1),想再次加入cuda的

conda install cudatoolkit=你所需要的版本(例如 cudatoolkit=10.2)

conda install cudnn=你所需要的版本(例如cudnn=7.6.5)

(3)# 有 CUDA 10.2 的, torch和cuda一起安装
conda install pytorch==1.7.0 torchvision==0.8.0 torchaudio==0.7.0 cudatoolkit=10.2 -c pytorch

测试pytorch安装成功的方法:

(1)在当前虚拟环境中输入python

(2)然后输入 import torch

3. cuda 10.2的下载与安装 

pytorch1.7.0版本的支持cuda10.2,所以本次下载的是cuda10.2

cuda下载地址选择cuda10.2进行下载

下载的很慢

先安装主要文件,cuda_10.2.89_441.22_win10

安装的时候要关掉360,避免安装失败。

选择自定义安装,只选第一个选项,其他默认即可。

安装完成后,在命令行窗口(cmd)中,输入 nvcc -V 命令进行测试。

 添加环境变量

右键单击此电脑——属性——高级系统设置——环境变量——系统变量

添加两个路径,以下是安装的路径(默认的):

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.2

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.2\lib\x64

4. cuDNN

cuDNN下载地址下载,据说cuda10.2和cudnn7.6.5搭配稳定可靠

 

 

 接下来的安装成功测试根据以下链接中的步骤来。

 检测是否安装成功,在文章的最后面

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值