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

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

安装tensorflow1.12版本

注意: 前面两行是下载文件到aistudio,后两行是安装,第一次需要先下载,往后如果还是进入到同样的项目环境中,那么就不用再执行前两行的语句了。

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-gpu == 1.12.0 keras==2.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.

发布了70 篇原创文章 · 获赞 0 · 访问量 3459
展开阅读全文

没有更多推荐了,返回首页

©️2019 CSDN 皮肤主题: 大白 设计师: CSDN官方博客

分享到微信朋友圈

×

扫一扫,手机浏览