如何在google colab创建/配置torch 0.3.1环境
背景
复现GAN论文,环境配置需要torch 0.3,但是我们的colab是最新的环境,这会导致某些预训练的包导入出错,下面实现怎么torch版本降级到torch 0.3,Colab的CUDA版本是11.6,而我需要9.0的CUDA版本,更改CUDA版本过程如下:
- 查看当前的cuda版本
# 1.先查看有无GPU
!nvcc --version
- 卸载当前Colab的CUDA
# 2.Uninstall the current CUDA version
!apt-get --purge remove cuda nvidia* libnvidia-*
!dpkg -l | grep cuda- | awk '{print $2}' | xargs -n1 dpkg --purge
!apt-get remove cuda-*
!apt autoremove
!apt-get update
- 查看对应Ubuntu版本,再去官网下载对应的cuda版本,可以先看一下Colab的系统版本,虽然我们的版本是Ubuntu 20的,但是我们下载是Ubuntu 16的版本,不影响的
# 查看版本
!lsb_release -a
然后知道系统版本,就可以去找对应的CUDA下载链接了,过程如下:
安裝cuda可以一步走完
# 3.下载对应的cuda版本
!wget https://developer.nvidia.com/compute/cuda/9.0/Prod/local_installers/cuda-repo-ubuntu1604-9-0-local_9.0.176-1_amd64-deb
!dpkg -i cuda-repo-ubuntu1604-9-0-local_9.0.176-1_amd64-deb
!apt-key add /var/cuda-repo-9-0-local/7fa2af80.pub
!apt-get update
!apt-get install cuda=9.0.176-1
!apt-get install cuda=9.0.176-1是强制系统安装CUDA9.0,一定要记得加,否则系统将会安装CUDA的最新版本!!!!
- 查看cuda版本
!nvcc -V
- 安裝cudnn
CUDA安装成功之后,再安装对应版本的cudnn,6.0之前的版本可以wget,7.0之後的版本要注册账号啥的,只能下載了文件《cudnn-9.0-linux-x64-v7.4.1.5.tgz》傳到google 云盘里面,然後使用下面命令安裝。
首先到官网,找到要下载的cudnn的版本
接下来上传到自己谷歌云盘,然后跳到该位置
# 需要自己提前下载好cudnn文件放在文件夹
# https://developer.nvidia.com/compute/machine-learning/cudnn/secure/v7.4.1.5/prod/9.0_20181108/cudnn-9.0-linux-x64-v7.4.1.5.tgz
%cd /content/drive/MyDrive/cudnn
进行下面的步骤安装
!sudo tar -zxvf cudnn-9.0-linux-x64-v7.4.1.5.tgz
!sudo cp cuda/include/cudnn.h /usr/local/cuda/include
!sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64
!sudo chmod a+r /usr/local/cuda/include/cudnn.h
!sudo chmod a+r /usr/local/cuda/lib64/libcudnn*
- 查看cudnn版本, 查看一下cudnn是否安装成功
!cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2
- 查看torch对应的python版本,click,下面安装python3.6的版本
%cd /content
# 下载python 3.6
%%bash
MINICONDA_INSTALLER_SCRIPT=Miniconda3-4.5.4-Linux-x86_64.sh
MINICONDA_PREFIX=/usr/local
wget https://repo.continuum.io/miniconda/$MINICONDA_INSTALLER_SCRIPT
chmod +x $MINICONDA_INSTALLER_SCRIPT
./$MINICONDA_INSTALLER_SCRIPT -b -f -p $MINICONDA_PREFIX
import sys
_ = (sys.path
.append("/usr/local/lib/python3.6/site-packages"))
- 下面安装对应的pytorch版本,click
# 下载
!pip install http://download.pytorch.org/whl/cu90/torch-0.3.1-cp36-cp36m-linux_x86_64.whl
!pip install torchvision==0.2.0
显示是否成功
!pip list
9. 最后进行你需要的依赖,然后就能开始跑预训练代码
10. 参考链接
https://www.twblogs.net/a/5efddbc25352062f754e3441
https://blog.csdn.net/qq_46521210/article/details/122211173
https://blog.csdn.net/yuezhilanyi/article/details/79470448
https://zhuanlan.zhihu.com/p/50302396
https://blog.csdn.net/YhL_Leo/article/details/50961542