2024pytorch小土堆安装教程(GPU),踩坑总结

目录

2024pytorch小土堆安装教程(GPU),踩坑总结

本人小白,应老师要求自学深度学习,跟着b站up小土堆老师学习深度学习,第一节课是安装pytorch,在历经三次失败后才获得成功,于是决心写下自己的总结以供大家参考借鉴。以下均基于小土堆的PyTorch深度学习快速入门教程的内容上得到,希望能对有同样问题的朋友们有所帮助。

一、安装Anaconda

这一步较简单,一般不会出问题。我在此之前学习数据预处理就已经安装好了,小土堆的视频有详细讲述如何下载Anaconda,此处不再赘述。
另外由于之前下载的Anaconda版本过低,在安装pytorch的时候提醒我安装,亲测不用更新版本也可以正确安装。
2024pytorch小土堆安装教程(GPU),踩坑总结_第1张图片

二、检查显卡及驱动安装

根据提供的三种方法查看电脑的显卡即可,若nvidia开头则后续pytorch安装中选择GPU版本即可,如无选择CPU版本。

三、创建虚拟环境

输入以下代码创建虚拟环境,-n代表名称
conda create -n 虚拟环境名称 python=版本
踩坑一:创建对应的python版本。小土堆的python版本为3.6,所以视频中他的代码为

conda create -n pytorch python=3.6 

你可能感兴趣的:(pytorch)