我有一个旧的conda虚拟环境,里面安装了pytorch和cudatoolkit 10.1。我能够运行代码,并且每次都会得到以下信息:
I tensorflow/stream_executor/platform/default/dso_loader.cc:48] Successfully opened dynamic library libcudart.so.10.1
然而,我使用cudatoolkit 11.0创建了一个新的虚拟环境,但现在代码开头不再运行这条消息(也没有出现未能找到文件的错误消息),并且当它尝试使用gpu时代码最终会崩溃。我该如何确保它正确加载了GPU支持?