记录cuda pytorch环境安装

最新推荐文章于 2025-08-12 13:36:56 发布

原创最新推荐文章于 2025-08-12 13:36:56 发布 · 721 阅读

1 ·

CC 4.0 BY-SA版权

文章标签：

#cuda #gpu #ubuntu

本文详细记录了在Ubuntu16.04环境下安装双版本CUDA(10.0与10.1)及对应PyTorch(1.0与1.5)的全过程，分享了从官网下载、配置到最终成功的经验，特别强调了利用清华镜像加速PyTorch安装的方法。

已经安装环境5，6次了，每次装好不做记录结果就忘了之后还得重查资料，所以这次决定做记录，以便以后查阅。
我现在使用的是ubuntu16.04，这次我要装得是双cuda+pytorch 已经装好一个版本的cuda10.0+pytorch1.0 现在开始装cuda10.1+pytorch1.5

步骤1、安装cuda10.1 参考教程https://blog.youkuaiyun.com/qlulibin/article/details/78714596）
现在官网下载cuda10.1的对应版本
https://developer.nvidia.com/cuda-toolkit-archive
步骤2、cuda多版本安装教程
参考链接 https://blog.youkuaiyun.com/u013066730/article/details/80984603
根据这两个链接，可以安装成功，出现其他问题可以再百度调整。

最后说一下pytorch安装最好挂一下清华镜像否则下载速度太慢。另外就是去pytorch官网找下载命令就好啦
pytorch安装
（1）输入清华镜像
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
conda config --set show_channel_urls yes
（2）然后pytorch官网找下载命令
例：conda install pytorch torchvision cudatoolkit=10.1
conda install这个命令不行，可以试试pip 这个命令的链接