colab完美配置深度学习环境
colab完美配置深度学习环境
配置python环境
一般情况下,colab的python环境是python3.7,而很多深度学习代码要求的环境有的可能比较老,比如python3.5。这个时候一定不要在原python环境中配置,如果这样做,大多数情况下都是配不好的。所以*步就是首先要配置要求的python环境。下面以python3.5举例;
首先安装python3.5
!at-get install python3.5
然后安装pip,不然无法在python3.5中使用pip安装包
!apt install python3-pip
接着,升级pip,不然有些包在安装时会出错
!python3.5 -m pip install –upgrade pip
*后,就是安装代码依赖的环境
!python3.5 -m pip install -r requirements.txt
普通安装包的话如下:
!python3.5 -m pip install tensorflow==1.1.0
配置CUDA环境
因为深度学习需要用gpu加速,如果你是使用colab自带的python环境的话,系统自带的cuda环境就是没问题的,可以直接用。但若是像上面一样配置python3.5的环境的话,cuda版本等都会出问题。所以这个时候就需要自己去配置cuda环境。