栏目分类:
子分类:
返回
名师互学网用户登录
快速导航关闭
当前搜索
当前分类
子分类
实用工具
热门搜索
名师互学网 > IT > 软件开发 > 后端开发 > Python

Anaconda+cuda+cudnn+Pytorch全家桶安装

Python 更新时间: 发布时间: IT归档 最新发布 模块sitemap 名妆网 法律咨询 聚返吧 英语巴士网 伯小乐 网商动力

Anaconda+cuda+cudnn+Pytorch全家桶安装

搭建环境:Windows 10 + Nvidia GTX1650显卡

花了整整两天时间,哥们终于把这仨玩意安装好了

anaconda

 

Anaconda是一个集成了python许多包的一个库吧,安装了anaconda就能等于是安装了Python

anaconda的安装并不复杂,在官网上找到下载安装即可

 

Python

要紧的是,anaconda可以创造多个环境,使不同的程序可以再不同的环境中运行

 

base环境与Pytorch环境

如上图所示,如果我们需要使用1.0版本的Pytorch,那我们只需在base环境下。 如果需要2.0的Pytorch,只需在Pytorch环境下。

打开cmd输入:

  1. 创建pytorch环境: conda create -n pytorch python=3.8.8

  2. 激活pytorch环境 : conda activate pytorch


CUDA

cuda的安装简直快给我累死了,先安装了10.2版本,后安装了11.4版本,最终选择了11.1版本

我列一列具体安装的几个坑吧:

Cuda安装时,Visual integration 报错,10.2和11.4皆遇到

 

解决方法:首先你必须要安装visual studio,比如我安装的是VS2019,遇到此问题大体是visual Studio的版本不对,此时只需要卸载visual studio,再安装,即可!

CUDA安装高版本后卸载安装低版本,发现系统仍显示高版本

以我自己举例,我安装了cuda11.4,卸载后重装了11.1版本,但是在系统输入:

nvcc -V后发现CUDA版本仍是11.4

 

解决方法:据我猜想哈,应该是系统环境变量没有更改,仍然是11.4,此时只需将所有包含CUDA的变量手动更改至自己版本号即可。

 

cudnn的安装

Cudnn安装十分简单,在Nivida官网上下载(要挂VPN),切记要安装与自己CUDA版本对应的CUDNN

 

想要安装则必须要登录Nvida账号

 

最新的cuda版本是11.4,我们的CUDA版本是11.1,所以我们选择历史下载

 

选择与自己版本号对应的cudnn

下载好了是一个压缩包,解压缩后得到三个文件夹

 

将这三个文件剪切至:C:Program FilesNVIDIA GPU Computing ToolkitCUDAv11.1

 

Pytorch的安装

打开Pytorch官网点击Install

 

选择自己的cuda版本,复制代码

 

复制 :conda install pytorch torchvision torchaudio cudatoolkit=11.1 -c pytorch -c conda-forge

在pytorch环境下输入

 稍作等待即可。

Pytorch安装失败

如果在安装过程中不小心关闭了,或者是第一次安装未成,进行第二次安装。此时会报错,显示有些文件已存在。

此时输入: conda clean -y -all 将所有的文件清空,再进行安装即可。

测试cuda是否可用

依次输入:

conda activate pytorch #激活pytorch环境

python #进入python

import torch #导入torch

torch.cuda.is_available() #查看是否含有cuda

 

返回的如果是TURE就大功告成!!

转载请注明:文章转载自 www.mshxw.com
本文地址:https://www.mshxw.com/it/294228.html
我们一直用心在做
关于我们 文章归档 网站地图 联系我们

版权所有 (c)2021-2022 MSHXW.COM

ICP备案号:晋ICP备2021003244-6号