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