Skip to content

forge本地搭建出现显卡CUDA Torch不兼容问题已解决 #2469

Answered by boboHenry
boboHenry asked this question in Q&A
Discussion options

You must be logged in to vote

第一步(安装你电脑的显卡驱动)

这个问题我解决了,首先如果你是Windows系统打开cmd,查看你的显卡版本信息(命令nvidia-smi此命令在你安装了你的显卡对应驱动后才可以使用),我的显卡版本是 Driver Version: 475.14。去nvidia官网https://www.nvidia.cn/drivers/lookup/下载对应的驱动版本。

第二步(下载安装你电脑显卡对应的CUDA)
CUDA下载地址https://developer.nvidia.com/cuda-toolkit-archive

安装好CUDA后使用nvcc -V查看安装是否成功以及对应的CUDA版本,我电脑CUDA安装的版本是11.4

第三步(安装PyTorch 地址:https://pytorch.org/get-started/previous-versions/)
不要安装最新版PyTorch会有兼容问题,要安装你电脑安装的CUDA版本对应的PyTorch,我这里安装的是# CUDA 11.6版本

以上三步就可以解决forge运行报显卡版本错误以及CUDA错误,以及CUDA对应的torch不兼容问题

Replies: 3 comments 3 replies

Comment options

You must be logged in to vote
0 replies
Comment options

You must be logged in to vote
3 replies
@boboHenry
Comment options

@boboHenry
Comment options

Answer selected by boboHenry
@boboHenry
Comment options

Comment options

You must be logged in to vote
0 replies
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Category
Q&A
Labels
None yet
2 participants