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

cuda鍜宑udnn瀹夎(anaconda cuda)

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

cuda鍜宑udnn瀹夎(anaconda cuda)

anaconda+ptyorch+cuda+cudnn安装

pytorch的实用向安装。简单明了,必成功。


下载与安装

anaconda+ptyorch+cuda+cudnn安装前言一、安装anaconda二、测试conda三、CUDA和cudnn支持四、cuda和cudnn安装

4.1 cuda4.2 cudnn 五、测试cuda和cudnn

5.1 cuda测试5.2 cudnn测试 六、pytorch-gpu下载七、pytorch-gpu安装八、测试pytorch-gpu总结


前言

昨晚,电脑重装。以往的程序代码工作全部白费。劝告各位一定要及时备份。
心痛不已。搞了一晚上安装。自己总结下以备下次使用。


一、安装anaconda

安装anaconda可以去官网安装:link

根据你的电脑选择你想下载的版本。我是Windows,然后64位。下载。

也可以使用百度云分享链接链接:https://pan.baidu.com/s/1IgRs9sfqifHwC7g3i7lx6g?pwd=e65r
提取码:e65r
安装就是一路next。


这个地方,all user就可以,自己的电脑嘛。

在安装路径那里一般为了以后省事。就默认安装就好。

再往下就是把第一个勾打上。

然后install 等待安装就好了。

二、测试conda

win+R 然后输入cmd。输入conda。


ok,这样就是完成了。继续下一步安装CUDA 和 CUDNN。

三、CUDA和cudnn支持

安装cuda和cudnn的话,你需要先查看自己的电脑是否支持。
首先打开NVIDIA控制面板,一般桌面鼠标右键就可以。
打开界面如下。点击左下角。

然后显示,组件两个,点击组件就可以看到自己的CUDA版本。我的是11.1 。

四、cuda和cudnn安装 4.1 cuda

如果你的电脑显示你有CUDA,那么你就可以继续安装啦。
我们打开pytorch官网
查看版本对应。
(这里。。嗯。。不看也行,知道自己CUDA版本直接往下看就好)

CUDA官网下载链接
这里我也放上我的云盘链接。版本为CUDA11.1。两者选一个下载就好。
链接:https://pan.baidu.com/s/1Fow38_usxMru2ualcrYUog?pwd=kcm3
提取码:kcm3

下载完成后,安装。
这里选择默认路径。

继续

精简就可以。

然后,继续安装,等待。

安装完成

4.2 cudnn

依然还是一个官网链接。cdunn官网链接
一个云盘链接。链接:https://pan.baidu.com/s/1HK4ZlSAMQMJdFZQwIoEbcQ?pwd=odn0
提取码:odn0

安装好了解压。会有三个文件。
将他们全选复制。然后打开CUDA刚才的安装路径。如果你是默认安装,路径应该和我一样。
将复制的三个文件,粘贴,替换。

五、测试cuda和cudnn 5.1 cuda测试

win+R ,cmd打开,输入 :cd C:Program FilesNVIDIA GPU Computing ToolkitCUDAv11.1bin
然后输入 :nvcc -V
出现下面即为成功。

5.2 cudnn测试

测试cudnn是否安装成功
win+R ,cmd打开,输入 :
cd C:Program FilesNVIDIA GPU Computing ToolkitCUDAv11.1extrasdemo_suite
输入:bandwidthTest.exe
显示PASS则安装成功

输入:deviceQuery.exe
若显示PASS,则cuda和cudnn均安装成功

六、pytorch-gpu下载

进入pytorch网站: https://pytorch.org/get-started/locally/
选择pip下载,进入下方网址,

这个地方下载,我的网络是不支持。太卡了。
所以我们可以转到清华镜像网站
也可以从云盘链接下载,第一个是pytorch,第二个是torchvision。
链接:https://pan.baidu.com/s/1Di3uNRq2C9J-usf7YTJyKA?pwd=l29y
提取码:l29y
链接:https://pan.baidu.com/s/1OfTst0iShpxtedWVHyJgqw?pwd=ttsu
提取码:ttsu

七、pytorch-gpu安装

打开刚才下载好的anaconda的 anaconda prompt

创建一个pytorch-gpu的环境

conda create --name pytorch-gpu python=3.7
conda activate pytorch-gpu

然后将目录转到你刚才下载的两个pytorch文件下。

输入以下命令分别安装两个包:记得安装时复制名字,要把bz2也加上。

conda install --offline pytorch-1.10.1-py3.7_cuda11.1_cudnn8_0.tar.bz2 #文件全称
conda install --offline torchvision-0.10.1-py37_cu111.tar.bz2 #文件全称

我已经安装好了。安装大概几分钟就好、速度很快。

然后继续 安装

conda install cudatoolkit=11.1

到这里就完成了。

八、测试pytorch-gpu

我们在环境下输入python 进入 python编译。然后输入测试代码。

import torch
print(torch.__version__)
print(torch.cuda.is_available())
print(torch.cuda_version)


搞定收工。

接下来也可以和pycharm链接。

总结

到这里就完美结束。

主要参考:
1.https://blog.csdn.net/qq_39022478/article/details/103938392?utm_medium=distribute.pc_relevant.none-task-blog-2%7Edefault%7EBlogCommendFromMachineLearnPai2%7Edefault-1.control&depth_1-utm_source=distribute.pc_relevant.none-task-blog-2%7Edefault%7EBlogCommendFromMachineLearnPai2%7Edefault-1.control
2.https://blog.csdn.net/weixin_44398263/article/details/107709934?utm_medium=distribute.pc_aggpage_search_result.none-task-blog-2allsobaiduend~default-3-107709934.nonecase&utm_term=pytorch%E6%89%80%E6%9C%89%E7%89%88%E6%9C%AC%E9%95%9C%E5%83%8F&spm=1000.2123.3001.4430

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

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

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