最近几天都在安装这两个环境，但由于是在服务器上安装，多少还是有些不方便。根据NVIDIA-smi，安装了cuda10.2，之后安装了对应的cudnn，安装对应的pytorch。
import torch
查了半天，大概意思环境不匹配，所以又装了cuda10.1.还是不行。走投无路看到了一篇博客(49条消息) TensorFlow/PyTorch和cuda等版本对应关系_gyxx1998的博客-CSDN博客_pytorch和cuda对应版本
试着装一下TensorFlow，结果很顺利就安装好了。非常感谢这位博主解救了我的周末。
安装TensorFlow的可以按照这篇博文就可以。
想着pytorch应该也可这样安装，于是打开pytorch主页，发现了这句：install previous version of pytorch
打开后找到对应的版本就可以在你创建的虚拟环境中安装了。感谢强大的anaconda！！！
步骤：（1）conda create -n envs-name python=3.8 创建虚拟环境
(2) conda activate envs-name 激活虚拟环境
(3)打开pytorch官网，点击install previous version of pytorch
（4）conda install pytorch torchviudio cudatoolkit=10.2 -c pytorch 安装pytorch，顺便把它需要的其他东东都自动帮你安好，再不用自己安装cuda和cudnn了。
（5）坐等安装
（6）python
import torch
…紧张的等待，没有error
（7）torch.cuda.is_available()
出现True
就成功啦！！！
花了3天没搞定，这个方法5分钟就安好了。
希望大家不要走弯路，浪费时间。
[En]
I hope everyone will not take detours and waste time.
Original: https://blog.csdn.net/u013955915/article/details/123433160
Author: 绿色生命
Title: ubuntu下安装pytorch和TensorFlow
原创文章受到原创版权保护。转载请注明出处：https://www.johngo689.com/509244/
转载文章受原作者版权保护。转载请注明原作者出处！