如何在google colab创建/配置torch 0.3.1环境

最新推荐文章于 2023-08-09 19:31:46 发布

yzLee12138

最新推荐文章于 2023-08-09 19:31:46 发布

阅读量1.7k

点赞数 7

文章标签： python 深度学习人工智能

本文链接：https://blog.csdn.net/m0_46550089/article/details/129119040

版权

如何在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