安装好anaconda后(我装的anaconda的base的python版本是3.8的,但没影响)
安装tensorflow-gpu
conda create -n tensorflow python=3.7 conda activate tensorflow conda install tensorflow-gpu
就好了
这里默认的是安装tensorflow2.0
安装tensorflow1.0conda create -n tensorflow1.0 python=3.6 conda activate tensorflow1.0 pip install tensorflow-gpu==1.13.1 -i https://pypi.tuna.tsinghua.edu.cn/simple
这里要注意tensorflow的版本要与安装的cuda10.0对应
最开始我安装了tensorflow-gpu=1.4.0和1.12.0都不行,安装完后倒入报错,一查发现它们只支持cuda9.0
注意安装tensorflow1.0的话要pyhon3.6或3.5,3.7不行
安装pytorch-gpu
conda create -n pytorch python=3.7 conda activate pytorch pip install torch==1.6.0 torchvision==0.7.0 -f http://download.pytorch.org/whl/torch_stable.html
下的比较慢,当时是当它下了一晚上
安装完之后就好了
ubuntu安装tensorflow-gpu和pytorch-gpu
最新推荐文章于 2022-10-30 12:51:14 发布