欢迎光临散文网 会员登陆 & 注册

40系显卡深度学习配置指南(pytorch)

2023-03-04 15:46 作者:开奔驰买油条  | 我要投稿

众所周知,每一代新显卡都会与指定的cuda新版本唯一兼容,例如30系显卡必须配合cuda11.X版本,40系也不例外,有独特的cuda版本需求。

40系目前兼容的最早cuda版本为11.8,见以下截图:

更早的cuda也能跑程序,但不保证效率

因此需要配置的cuda版本最早为11.8,下载的pytorch也需要与这个版本对应,官网的下载方式如下截图:

哎呀,我的cuda11.8呢?

截图截错了?我们看看另一边的Preview呢:

一眼丁真!

好的,在preview里找到了。下载好如上版本的pytorch,cuda后即可配对,实现40系显卡的深度学习配置环境。如果需要cudnn的同学,下载cudnn对应的11.x版本即可。

cuda下载直链:https://developer.download.nvidia.com/compute/cuda/11.8.0/local_installers/cuda_11.8.0_522.06_windows.exe
pytorch下载直链(pip格式):pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu118

40系显卡深度学习配置指南(pytorch)的评论 (共 条)

分享到微博请遵守国家法律