直接上报错的图:
前提条件:检查了cuda.is_available()和CUDA_HOME为True
解决方案:conda install -c pytorch pytorch-nightly cuda100 (我的cuda为cuda10.0)
conda install -c pytorch pytorch-nightly cuda100 (我的cuda为cuda10.0)
安装请重新编译