深度学习框架gpu安装方法

时间:2022-01-26 19:53:40

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/