方法1:由于anaconda环境中存在两个 libiomp5md.dll文件,
其中一个是torch路径下的,另一个是虚拟环境本身路径中的,转到第二个目录下把它剪切到其他路径下备份就好(最好把路径也备份一下)。
方法2:程序开头放置以下代码
import os
os.environ["CUDA_VISIBLE_DEVICES"] = "0"
方法1:由于anaconda环境中存在两个 libiomp5md.dll文件,
其中一个是torch路径下的,另一个是虚拟环境本身路径中的,转到第二个目录下把它剪切到其他路径下备份就好(最好把路径也备份一下)。
方法2:程序开头放置以下代码
import os
os.environ["CUDA_VISIBLE_DEVICES"] = "0"