较新的显卡用了CUDA 12.1版本,如何获取最新的pytorch版本GPU上运行

前往官网:https://pytorch.org/get-started/locally/

点击你想使用的平台和安装方法
在主页上点击你想使用的平台和安装方法, 得到表格最下面的:
Run this Command:的建议:

conda install pytorch torchvision torchaudio pytorch-cuda=11.8 -c pytorch -c nvidia

如何开始安装?

在我另一篇文章中提到的,安装好cuda和conda后,
在conda命令终端中,输入以下指令,建立新环境

conda create -n 【name_of_new_environment】python=【python_version】

也就是:
conda create -n leer python = 3.10.10
那么就会有d:\Leer\envs\leer\这个目录出现。可以cd进去。
然后运行安装命令:
conda install pytorch torchvision torchaudio pytorch-cuda=11.8 -c pytorch -c nvidia
如图:
在这里插入图片描述
中途点击 “y” 确认开始执行。
然后等待完成,等待过程中,这篇文章诞生了。
更多细节参考文章:
https://blog.csdn.net/qq_43855258/article/details/128690570
https://blog.csdn.net/AiTanXiing/article/details/129933911

Logo

昇腾计算产业是基于昇腾系列(HUAWEI Ascend)处理器和基础软件构建的全栈 AI计算基础设施、行业应用及服务,https://devpress.csdn.net/organization/setting/general/146749包括昇腾系列处理器、系列硬件、CANN、AI计算框架、应用使能、开发工具链、管理运维工具、行业应用及服务等全产业链

更多推荐