文章详情

短信预约-IT技能 免费直播动态提醒

请输入下面的图形验证码

提交验证

短信预约提醒成功

CUDA+Pytorch安装教程(包含踩坑内容)

2023-09-04 20:14

关注

第一次写此类文章,有写得不好的地方还请指出,下面进入正题。

1.CUDA安装及卸载

安装

在安装CUDA前,需要先确定自己电脑是否有独显,设备管理器-->显示适配器:

 一般性都是NVIDIA开头的为独显。

查看电脑支持的最高CUDA版本,打开任务栏-->隐藏-->NVIDIA图标:

 选择帮助-->系统信息-->组件:

在NVCUDA.DLL行可以看到我的电脑最高能带动NVCUDA10.2,一般性这样查看就可以了。另一种方法是在显示中查看驱动程序版本,此处不做详细介绍。

去官网下载对应的CUDA版本,此处选版本没必要特地降版本,官网地址:

CUDA Toolkit Archive | NVIDIA Developericon-default.png?t=N2N8https://developer.nvidia.com/cuda-toolkit-archive打开安装包,第一步无需更改,直接ok:

 选择自定义安装,如果第一次安装CUDA,四个选项都需勾选,如果不是第一次安装,建议只勾选第一个选项(CUDA)。在CUDA选项中将Visual Studio Integration选项去掉,如下:

 

后序的安装位置推荐可以将C盘改为D盘,但C盘后面的安装路径需要原封不动,如下展示:

之后不做任何更改。

查看是否安装成功,可以打开命令指示符,输入nvcc -V,有如下图标便为成功:

卸载

控制面板-->程序-->程序和功能,找到NVIDIA的位置所在:

 除了NVIDIA GeForce Experience,NVIDIA PhysX 系统软件,NVIDIA 图形驱动程序不卸载,别的带NVIDIA开头的都卸载。 

2.Pytorch安装

 以上步骤完成后,电脑已经有了对应的CUDA。

注意,Python,CUDA和Pytorch都需安装对应版本,查看对应版本我看的是这篇文章:

https://blog.csdn.net/weixin_45508265/article/details/122006134

先下载好对应的Python版本,Anconda3。

打开Anconda3的Anconda Prompt,输入指令:conda create -n pytorch python=3.7,这里的pytorch为虚拟环境名字,python=3.7为我的电脑的python版本,请选择自己电脑的版本安装。然后输入指令:conda activate pytorch,进入相应的虚拟环境。

输入python -V查看此环境版本号,输入nvidia-smi查看CUDA版本号:

 python版本和CUDA版本匹配后再进行后续操作。

由于默认路径下载pytorch太慢,需更换为清华源路径。

输入:conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/即可切换到清华源路径,然后去官网下载pytorch:

PyTorchicon-default.png?t=N2N8https://pytorch.org/选择Stable+Windows+Conda+Python。

我选择的版本是Python3.7+Pytorch1.7.1+CUDA10.2,因此安装指令是:

此处不要添加-c pytorch。添加-c pytorch后会使下载路径变为默认,导致下载速度特别慢。

3.Pytorch安装过程问题解决

 如果按照以上操作中间出现问题并且网上别的方法无法解决的时候,可以尝试重启电脑,如果也不行,可以尝试使用如下方法:

第一步是删除相应的python虚拟环境,并重新建一个,此步无需安装pytorch。

第二步是删除ANCONDA3-->pkgs中的所有文件。在pkgs中保存是安装的所有外来包。

第三步在c盘中搜索.condarc。此文件存放着下载路径,如下所示:

 可以更改成我所设置的参数:

ssl_verify: true
show_channel_urls: true
channels:
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
  - http://mirrors.ustc.edu.cn/anaconda/pkgs/main/
第四步:在第一步新创的环境中安装pytorch。

 

来源地址:https://blog.csdn.net/weixin_48556553/article/details/129852474

阅读原文内容投诉

免责声明:

① 本站未注明“稿件来源”的信息均来自网络整理。其文字、图片和音视频稿件的所属权归原作者所有。本站收集整理出于非商业性的教育和科研之目的,并不意味着本站赞同其观点或证实其内容的真实性。仅作为临时的测试数据,供内部测试之用。本站并未授权任何人以任何方式主动获取本站任何信息。

② 本站未注明“稿件来源”的临时测试数据将在测试完成后最终做删除处理。有问题或投稿请发送至: 邮箱/279061341@qq.com QQ/279061341

软考中级精品资料免费领

  • 历年真题答案解析
  • 备考技巧名师总结
  • 高频考点精准押题
  • 2024年上半年信息系统项目管理师第二批次真题及答案解析(完整版)

    难度     807人已做
    查看
  • 【考后总结】2024年5月26日信息系统项目管理师第2批次考情分析

    难度     351人已做
    查看
  • 【考后总结】2024年5月25日信息系统项目管理师第1批次考情分析

    难度     314人已做
    查看
  • 2024年上半年软考高项第一、二批次真题考点汇总(完整版)

    难度     433人已做
    查看
  • 2024年上半年系统架构设计师考试综合知识真题

    难度     221人已做
    查看

相关文章

发现更多好内容

猜你喜欢

AI推送时光机
位置:首页-资讯-后端开发
咦!没有更多了?去看看其它编程学习网 内容吧
首页课程
资料下载
问答资讯