如何在百度AI Studio上安装tensorflow和pytorch

时间:2024-04-08 17:48:26

安装的当然得是GPU版本,不然多浪费显卡

安装tensorflow1.12版本

wget https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/linux-64/cudatoolkit-9.0-h13b8566_0.conda -P pypi2/

python3.6 -m pip download tensorflow-gpu == 1.12.0 keras==2.1.5 -i https://pypi.tuna.tsinghua.edu.cn/simple -d pypi2

conda install --use-local pypi2/cudatoolkit-9.0-h13b8566_0.conda -c https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/

python3.6 -m pip install tensorflow-gpu1.12.0 keras2.1.5 --no-index -f ./pypi2

如何在百度AI Studio上安装tensorflow和pytorch
接下来就使用每天免费12小时的V100吧

另外注意由于这里不是在默认的python3.7环境下,而是在python3.6环境下安装的tensorflow,所以运行代码时要用python3.6 xx.py. 而且在安装其它包的时候也要用python3.6 -m pip install xx

安装pytorch gpu

建议将下面几行代码写成python脚本文件,例如installtorch.py
import os
if not os.path.exists(‘pypi’):
os.mkdir(‘pypi’)
if not ‘lxml’ in ’ ‘.join(os.listdir(‘pypi/’)):
os.system(‘pip download bs4 lxml xlwt xlrd seaborn -i https://pypi.tuna.tsinghua.edu.cn/simple -d pypi’)
if not ‘torch’ in ’ ‘.join(os.listdir(‘pypi/’)):
os.system(‘pip download torch1.4.0+cu92 torchvision0.5.0+cu92 -f https://download.pytorch.org/whl/torch_stable.html -d pypi’)
if os.popen(‘pip show torch’).read()==’’ and ‘torch’ in ’ '.join(os.listdir(‘pypi/’)):
os.system(‘pip install bs4 lxml xlwt xlrd seaborn torch torchvision --no-index -f ./pypi’)

然后在终端执行
python installtorch.py
接下来就会安装了
如何在百度AI Studio上安装tensorflow和pytorch

另外AI studio上的vim tab键默认是缩进8个空格,虽然不会报错,但是看着很别扭,可以进入命令行模式(esc键)然后

:set tabstop=4

回车就可以看到正常的缩进,但是这是临时的,每次进入vim后又是默认的8个缩进,然而AI studio没有给你sudo权限,你就没法更改/etc/vim/vimrc来配置vim.