ubuntu16.04+cuda8.0+cudnn6.0.21安装pytorch的gpu版本

小企鹅爱吃鱼

于 2020-09-24 16:22:54 发布

阅读量279

点赞数

分类专栏：环境配置文章标签： linux ubuntu

本文链接：https://blog.csdn.net/yinge1523/article/details/108776294

版权

环境配置专栏收录该内容

2 篇文章 0 订阅

订阅专栏

环境配置

1、版本问题
2、配置.profile .bashrc

1、版本问题

因为需要用都学校服务器，所以研究了环境配置，过程真是很曲折啊
服务器上的环境设置ubuntu16.04+cuda8.0+cudnn6.0.21
官网上关于之前版本（cuda8.0的找到了一个可以用的）
conda install pytorch1.0.0 torchvision0.2.1 cud80 -c pytorch
因为我在博客上看的，没有设置版本，导致下载的pytorch版本过高，导致cuda 用不了
下面是版本要求，我忘了是在哪里借鉴的图片。
在这里插入图片描述

安装过程有很多，自己找找看，包括镜像啊，离线安装呀，自己配置cuda，cudnn等（我真的是各种方式都试过）

2、配置.profile .bashrc

在版本没问题之后，python执行不了，看了很多解决方式，但是说的很笼统，就是那种今天可以，明天重新连的时候，又不行了。反正就是环境变量找不到，对于我们这种linux新手来说，真是太痛苦。（因为我的文件没有放在用户家目录下，但是PATH都指向那里，开始没注意这个问题，所以出了很多问题）
~是进入当前用户主目录
在这里插入图片描述
正常安装Anaconda后，在最后配置环境变量的需要三步：
1、vim /sdc/yanqilong/.bash_profile ,然后加入下面语句：

export PATH=/sdc/yanqilong/anaconda3/bin/:$PATH       
 #这是安装时的默认路径，如果安装的时候自定义了安装路径则需要更改，下同

/sdc/yanqilong/改为自己的用户路径就可以
2、vim /sdc/yanqilong/.bashrc ,然后加入下面语句：

# !! Contents within this block are managed by 'conda init' !!
__conda_setup="$('/sdc/yanqilong/anaconda3/bin/conda' 'shell.bash' 'hook' 2> /dev/null)"
if [ $? -eq 0 ]; then
    eval "$__conda_setup"
else
    if [ -f "/sdc/yanqilong/anaconda3/etc/profile.d/conda.sh" ]; then
        . "/sdc/yanqilong/anaconda3/etc/profile.d/conda.sh"
    else
        export PATH="/sdc/yanqilong/anaconda3/bin:$PATH"
    fi
fi
unset __conda_setup
export PATH="/sdc/yanqilong/anaconda3/bin:$PATH"
# >>> conda initialize >>>