最近几天都在安装这两个环境,但由于是在服务器上安装,多少还是有些不方便。根据NVIDIA-smi,安装了cuda10.2,之后安装了对应的cudnn,安装对应的pytorch。
import torch
出现了OSerror,
查了半天,大概意思环境不匹配,所以又装了cuda10.1.还是不行。走投无路看到了一篇博客(49条消息) TensorFlow/PyTorch和cuda等版本对应关系_gyxx1998的博客-优快云博客_pytorch和cuda对应版本
试着装一下TensorFlow,结果很顺利就安装好了。非常感谢这位博主解救了我的周末。
安装TensorFlow的可以按照这篇博文就可以。
想着pytorch应该也可这样安装,于是打开pytorch主页,发现了这句:install previous version of pytorch
打开后找到对应的版本就可以在你创建的虚拟环境中安装了。感谢强大的anaconda!!!