AntonioSu

1.tensorflow

pip install tensorflow-gpu==1.14.0,具体安装哪一个版本,可以把1.14.0随便填写一个数字,系统会提示可以有哪些版本可以安装

2.pytorch

conda install pytorch torchvision cudatoolkit=9.0 -c pytorch,这样会安装cuda版本是9.0对应的pytorch

 

注意事项:

通过pip安装python包时,确定不同的python对应的pip,比如:python3.5 需要使用pip3.5

对于同一个系统既需要TensorFlow有需要pytorch,可以先安装TensorFlow,TensorFlow对应的numpy会满足pytorch,如果numpy不满足,需要更新版本。

对于同一系统中安装了多个cuda版本,如果想使用某一版本,参考https://www.cnblogs.com/yhjoker/p/10972795.html

TensorFlow和keras对应关系: https://docs.floydhub.com/guides/environments/

分类:

技术点:

相关文章:

  • 2021-05-25
  • 2021-04-20
  • 2021-04-20
  • 2021-05-14
  • 2021-04-23
  • 2021-08-23
  • 2022-01-10
猜你喜欢
  • 2021-09-28
  • 2021-08-28
  • 2021-06-26
  • 2021-11-15
  • 2022-01-30
  • 2021-09-06
  • 2021-06-11
相关资源
相似解决方案