深度学习框架gpu安装方法

流过昼夜 提交于 2019-11-30 19:54:00

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

 

易学教程内所有资源均来自网络或用户发布的内容,如有违反法律规定的内容欢迎反馈
该文章没有解决你所遇到的问题?点击提问,说说你的问题,让更多的人一起探讨吧!