python 属性错误：模块"torch._C"没有属性"_cuda_setDevice"

yrwegjxp 于 2023-01-16 发布在 Python

关注(0)|答案(1)|浏览(1422)

我正在使用以下链接https://github.com/neuralchen/SimSwap探索SimSwap，但遇到了一个问题，即AttributeError：模块“torch._C”在运行以下命令时没有属性“_cuda_setDevice”！cd SimSwap & python test_one_image.py --人名--弧路径弧面模型/弧面检查点. tar--pic_a_路径crop_224/2. jpg--pic_b_路径crop_224/trump.jpg--输出路径输出/
在某个地方我发现这是cuda pytorch的一个问题，并发现以下代码
康达安装pytorch Torch 视觉 Torch 音频pytorch-cuda=11.6 -c pytorch -c英伟达
但我不是为自己工作。

python

来源：https://stackoverflow.com/questions/75125621/attributeerror-module-torch-c-has-no-attribute-cuda-setdevice

1条答案

按热度按时间

nhhxz33t1#

当PyTorch和CUDA版本不兼容时，通常会弹出此错误。

conda list cudatoolkit
conda list torch

你可以试着卸载两个库并重新安装它们.或者用正确的版本创建一个新环境

conda create --name cudatorchenv pytorch=X.X cudatoolkit=X.X

从https://discuss.pytorch.org/t/is-there-a-table-which-shows-the-supported-cuda-version-for-every-pytorch-version/105846开始
您可以从任何CUDA版本9.2的源代码构建PyTorch，二进制文件随install selection of PyTorch中提到的CUDA版本一起提供

赞(0）回复(0）举报 2023-01-16

我来回答

python 属性错误：模块"torch._C"没有属性"_cuda_setDevice"

1条答案

相关问题

热门标签

最新问答