问题描述
python=3.6,安装了torch=1.9,torch-scatter,但import torch的时候报错。网上说dll文件日期冲突,然并没有找到相关文件。后来用conda重装的torch(之前用pip装的),问题解决
开始时以为conda是10.2,后来出现torch和cuda版本不一致的问题。查看cuda版本,在官网上选定后重新安装torch即可。
解决办法
先在pycharm的terminal终端里用pip卸载torch,
pip uninstall torch
接着cmd中查看系统的cuda版本,
nvcc --version
我的显示的是11.1版本。
点进torch官网PyTorch,选定对应版本(conda、CUDA11.1),复制最下方的命令,粘贴到pycharm的terminal终端。
conda install pytorch torchvision torchaudio cudatoolkit=11.1 -c pytorch -c conda-forge
安装需要大概15-20分钟,差不多3个G,
完成。
我的项目还需要安装torch-scatter依赖包,terminal中用pip命令安装即可:
pip install --verbose --no-cache-dir torch-scatter