pytorch无法使用cuda

pytorch已经成功安装,是按照官网的conda install pytorch torchvision cudatoolkit=10.0 -c pytorch
cuda和cudnn也装了,通过nvcc指令可以看到是 Cuda compilation tools, release 10.0, V10.0.130
但是执行torch.cuda.is_available()时结果是False,请问大家要怎么解决呢

慕桂英3389331
浏览 17456回答 3
3回答

Kevin长亭

安装cuda10.1的版本即可

萧十郎

是不是创建了virtualenv环境
打开App,查看更多内容
随时随地看视频慕课网APP

相关分类

Python