安装的当然得是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
接下来就使用每天免费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上的vim tab键默认是缩进8个空格,虽然不会报错,但是看着很别扭,可以进入命令行模式(esc键)然后
:set tabstop=4
回车就可以看到正常的缩进,但是这是临时的,每次进入vim后又是默认的8个缩进,然而AI studio没有给你sudo权限,你就没法更改/etc/vim/vimrc来配置vim.