我的编程空间,编程开发者的网络收藏夹
学习永远不晚

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

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

北京

  • 北京
  • 上海
  • 天津
  • 重庆
  • 河北
  • 山东
  • 辽宁
  • 黑龙江
  • 吉林
  • 甘肃
  • 青海
  • 河南
  • 江苏
  • 湖北
  • 湖南
  • 江西
  • 浙江
  • 广东
  • 云南
  • 福建
  • 海南
  • 山西
  • 四川
  • 陕西
  • 贵州
  • 安徽
  • 广西
  • 内蒙
  • 西藏
  • 新疆
  • 宁夏
  • 兵团
手机号立即预约

请填写图片验证码后获取短信验证码

看不清楚,换张图片

免费获取短信验证码

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

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

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

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

下载Word文档到电脑,方便收藏和打印~

下载Word文档

编程热搜

  • Python 学习之路 - Python
    一、安装Python34Windows在Python官网(https://www.python.org/downloads/)下载安装包并安装。Python的默认安装路径是:C:\Python34配置环境变量:【右键计算机】--》【属性】-
    Python 学习之路 - Python
  • chatgpt的中文全称是什么
    chatgpt的中文全称是生成型预训练变换模型。ChatGPT是什么ChatGPT是美国人工智能研究实验室OpenAI开发的一种全新聊天机器人模型,它能够通过学习和理解人类的语言来进行对话,还能根据聊天的上下文进行互动,并协助人类完成一系列
    chatgpt的中文全称是什么
  • C/C++中extern函数使用详解
  • C/C++可变参数的使用
    可变参数的使用方法远远不止以下几种,不过在C,C++中使用可变参数时要小心,在使用printf()等函数时传入的参数个数一定不能比前面的格式化字符串中的’%’符号个数少,否则会产生访问越界,运气不好的话还会导致程序崩溃
    C/C++可变参数的使用
  • css样式文件该放在哪里
  • php中数组下标必须是连续的吗
  • Python 3 教程
    Python 3 教程 Python 的 3.0 版本,常被称为 Python 3000,或简称 Py3k。相对于 Python 的早期版本,这是一个较大的升级。为了不带入过多的累赘,Python 3.0 在设计的时候没有考虑向下兼容。 Python
    Python 3 教程
  • Python pip包管理
    一、前言    在Python中, 安装第三方模块是通过 setuptools 这个工具完成的。 Python有两个封装了 setuptools的包管理工具: easy_install  和  pip , 目前官方推荐使用 pip。    
    Python pip包管理
  • ubuntu如何重新编译内核
  • 改善Java代码之慎用java动态编译

目录