1.安装anaconda

官网
image.png
image.png
安装好后打开终端
image.png

  1. #新建名叫TF2.1的环境用python3.9版本
  2. conda create -n TF2.1 python=3.9
  3. conda activate TF2.1
  4. #安装英伟达的SDK10.1版本
  5. conda install cudacoolkit=10.1
  6. #安装英伟达深度学习软件包
  7. conda install cudnn=7.6
  8. #若上面两条语句报错了,则是电脑不支持英伟达GPU,就可以跳过
  9. #安装tensorflow
  10. pip install tensorflow==2.1
  11. #验证
  12. python
  13. import tensorflow as tf
  14. tf.__version__
  15. #安装pytorch(cudacoolkit=10.1)
  16. conda install pytorch torchvision torchaudio cudatoolkit=10.1 -c pytorch
  17. #验证TF2.1下输入
  18. python
  19. import numpy
  20. import torch

image.pnghttps://pytorch.org/

2.Jupiter Notebook

为新的TF2.1环境添加Jupiter Notebook
image.png

3.与pycharm

破解:WX公众号:python极客专栏

image.pngidea破解