【pytorch】AI Studio配置pytorch环境

参考：https://blog.csdn.net/firesolider/article/details/104948008

1.前言

穷哈没有卡，学校硬件暂时还没跟上。只能从网上找方法了。尝试过：
①CoLab,失败，需翻墙。本人不会科学上网，哭唧唧，求带
②kaggle kernel注册失败，折腾了白天，哭唧唧×2
③AI Studio，有免费算力，但是只能用paddlepaddle。之前参加过一期百度paddle的训练营，对AI Studio还是很有好感的。支持国产,但无奈我的代码已经用pytorch完工了。

只能猥琐发育了，尝试下能不能在AI Studio配置pytorch环境。

2.配置方法

百度的AI studio真的是良心了，V100显卡，显存16G,免费的卡，虽然平台主张使用paddle paddle其实pytorch也是可以用的按道理tensorflow也是可以用，但是cuda出现一点问题，虽然可以配置pytorch环境但是每次重启服务器之后又要重新配置pytorch，如果要配置的环境比较多的话，安装时间也是需要考虑的问题，还是不太方便，我猜百度是随机分配服务器的，所以这只适合没钱买显卡的学生党，而且算是送福利了，没卡真的很伤！

百度的anaconda环境直接给你配好了，所以可以直接使用conda命令

1. 在终端使用conda创建环境

aistudio@jupyter-153627-474162:~$ conda create -n pytorch python=3.6
Fetching package metadata .............
Solving package specifications: .

Package plan for installation in environment /opt/conda/envs/pytorch:

The following NEW packages will be INSTALLED:

    _libgcc_mutex:    0.1-conda_forge              conda-forge
    _openmp_mutex:    4.5-0_gnu                    conda-forge
    bzip2:            1.0.8-h516909a_2             conda-forge
    ca-certificates:  2020.4.5.1-hecc5488_0        conda-forge
    certifi:          2020.4.5.1-py36hc560c46_0    conda-forge
    expat:            2.2.9-he1b5a44_2             conda-forge
    gdbm:             1.18-h0a1914f_1              conda-forge
    ld_impl_linux-64: 2.34-h53a641e_0              conda-forge
    libffi:           3.2.1-he1b5a44_1007          conda-forge
    libgcc-ng:        9.2.0-h24d8f2e_2             conda-forge
    libgomp:          9.2.0-h24d8f2e_2             conda-forge
    libstdcxx-ng:     9.2.0-hdf63c60_2             conda-forge
    ncurses:          6.1-hf484d3e_1002            conda-forge
    openssl:          1.1.1g-h516909a_0            conda-forge
    pip:              20.1-pyh9f0ad1d_0            conda-forge
    pypy3.6:          7.3.1-h3e02ecb_1             conda-forge
    python:           3.6.10-h8356626_1011_cpython conda-forge
    python_abi:       3.6-1_pypy36_pp73            conda-forge
    readline:         8.0-hf8c457e_0               conda-forge
    setuptools:       46.3.0-py36hc560c46_0        conda-forge
    sqlite:           3.30.1-hcee41ef_0            conda-forge
    tk:               8.6.10-hed695b0_0            conda-forge
    wheel:            0.34.2-py_1                  conda-forge
    xz:               5.2.5-h516909a_0             conda-forge
    zlib:             1.2.11-h516909a_1006         conda-forge

Proceed ([y]/n)? y

_libgcc_mutex- 100% |########################################################| Time: 0:00:00   9.58 kB/s
ca-certificate 100% |########################################################| Time: 0:00:00 172.43 kB/s
ld_impl_linux- 100% |########################################################| Time: 0:00:01 452.09 kB/s
libstdcxx-ng-9 100% |########################################################| Time: 0:00:02   1.60 MB/s
libgomp-9.2.0- 100% |########################################################| Time: 0:00:01 741.48 kB/s
_openmp_mutex- 100% |########################################################| Time: 0:00:00 513.51 kB/s
libgcc-ng-9.2. 100% |########################################################| Time: 0:00:03   2.32 MB/s
bzip2-1.0.8-h5 100% |########################################################| Time: 0:00:00   3.33 MB/s
expat-2.2.9-he 100% |########################################################| Time: 0:00:00 617.72 kB/s
libffi-3.2.1-h 100% |########################################################| Time: 0:00:00  32.82 MB/s
ncurses-6.1-hf 100% |########################################################| Time: 0:00:00   1.50 MB/s
openssl-1.1.1g 100% |########################################################| Time: 0:00:01   1.67 MB/s
xz-5.2.5-h5169 100% |########################################################| Time: 0:00:00 836.30 kB/s
zlib-1.2.11-h5 100% |########################################################| Time: 0:00:00  42.31 MB/s
readline-8.0-h 100% |########################################################| Time: 0:00:00 813.40 kB/s
tk-8.6.10-hed6 100% |########################################################| Time: 0:00:01   2.03 MB/s
gdbm-1.18-h0a1 100% |########################################################| Time: 0:00:00 339.34 kB/s
sqlite-3.30.1- 100% |########################################################| Time: 0:00:01   1.77 MB/s
pypy3.6-7.3.1- 100% |########################################################| Time: 0:00:13   2.40 MB/s
python-3.6.10- 100% |########################################################| Time: 0:00:15   2.26 MB/s
python_abi-3.6 100% |########################################################| Time: 0:00:00   5.26 MB/s
certifi-2020.4 100% |########################################################| Time: 0:00:00 498.18 kB/s
setuptools-46. 100% |########################################################| Time: 0:00:00   1.58 MB/s
wheel-0.34.2-p 100% |########################################################| Time: 0:00:00  20.95 MB/s
pip-20.1-pyh9f 100% |########################################################| Time: 0:00:00   2.93 MB/s
#
# To activate this environment, use:
# > source activate pytorch
#
# To deactivate an active environment, use:
# > source deactivate
#

测试一下

aistudio@jupyter-153627-474162:~$ source activate pytorch
(pytorch) aistudio@jupyter-153627-474162:~$ source deactivate

成功

2. 给环境配置pytorch
去pytorch官网
选择版本：根据自己的需要选择。注意百度的cuda版本是9.2 注意的pytorch版本注意是否与cuda9.2匹配。当然也可以自己换cuda版本，有点麻烦并且每次重新配置环境不建议。

复制命令行执行就可以了。
pip install torch==1.5.0+cu92 torchvision==0.6.0+cu92 -f https://download.pytorch.org/whl/torch_stable.html

检查一下：

(pytorch) aistudio@jupyter-153627-474162:~$ pip list
Package     Version
----------- -------------------
certifi     2020.4.5.1
future      0.18.2
numpy       1.18.4
Pillow      7.1.2
pip         20.1
setuptools  46.3.0.post20200513
torch       1.5.0+cu92
torchvision 0.6.0+cu92
wheel       0.34.2

安装成功。

接下来就可以快乐的码代码啦~~~~
最后，强力给百度打 Call，阿中哥雄起