Linux 的 anaconda 虚拟环境下安装指定的 cuda、cudnn、pytorch
感悟:首先,anaconda 的虚拟环境真香!开辟一个新的虚拟环境,很多环境、版本不兼容的问题都不复存在,尤其对复现别人代码的同学很有用。
条件:只要安装的版本不超过自己机器的硬件条件,那么就可以安装。
步骤:
1. 确定安装的 cuda 版本。
在虚拟环境中,先用 conda search cudatoolkit --info 命令查看源内所有的 cuda 版本,以及下载地址。下图示例中展红框标出了支持的 cuda 版本、对应的 url 地址及一些约束条件。
2. 下载并安装 cuda。
找到你想要的且满足自己机器条件的 cuda 版本,复制 url 对应的下载链接,cd 到你想要的下载目录,用如下代码下载:
wget 复制的 url 链接
执行如下命令安装 cuda。如不在当前目录下,需在 cuda 包前加上绝对路径。
conda install 刚下载的 cuda 安装包
3. 确定 cuda 对应的 cudnn 版本。
用如下命令查看 conda 支持的 cudnn 版本,需要注意的是 cudnn 的版本应与 cuda 版本匹配。下图中标出了 cudnn 的版本号、url 下载地址及该版本对应的一些约束条件。
conda search cudnn --info
4. 下载并安装 cudnn。
找到你想要的且满足自己机器条件的 cudnn 版本,复制 url 对应的下载链接,用如下代码下载:
wget 复制的 url 链接
执行如下命令安装 cudnn。
conda install 刚下载的 cudnn 安装包
5. 安装 pytorch。
在官网上找到安装的 cuda 版本所支持的 pytorch 版本后,复制类似于下图红框中的命令至Linux命令行执行。
conda install pytorch==1.4.0 torchvision==0.5.0 cudatoolkit=10.1 -c pytorch
5. 测试安装是否成功。
在虚拟环境中测试 cuda 是否安装成功不能用 nvcc -V。这是因为该命令返回的系统默认的 cuda 版本,而非当前环境下使用的 cuda 版本。我们可以在 python 环境下用 pytorch 命令检查当前环境下执行 python 文件时调用的是哪个 cuda。
# 命令行进入 python 交互环境后执行如下命令
import torch # 若不报错则 pytorch 安装成功,否则失败。
torch.__version__ # 查看 torch 版本
torch.version.cuda() # 若 cuda 安装成功则返回版本号,否则失败。
torch.backends.cudnn.version() # 若 cudnn 安装成功则返回版本号,否则失败。
这篇好文章是转载于:学新通技术网
- 版权申明: 本站部分内容来自互联网,仅供学习及演示用,请勿用于商业和其他非法用途。如果侵犯了您的权益请与我们联系,请提供相关证据及您的身份证明,我们将在收到邮件后48小时内删除。
- 本站站名: 学新通技术网
- 本文地址: /boutique/detail/tanhgcfjac
同类精品
更多
-
photoshop保存的图片太大微信发不了怎么办
PHP中文网 06-15 -
Android 11 保存文件到外部存储,并分享文件
Luke 10-12 -
《学习通》视频自动暂停处理方法
HelloWorld317 07-05 -
word里面弄一个表格后上面的标题会跑到下面怎么办
PHP中文网 06-20 -
photoshop扩展功能面板显示灰色怎么办
PHP中文网 06-14 -
微信公众号没有声音提示怎么办
PHP中文网 03-31 -
excel下划线不显示怎么办
PHP中文网 06-23 -
怎样阻止微信小程序自动打开
PHP中文网 06-13 -
excel打印预览压线压字怎么办
PHP中文网 06-22 -
TikTok加速器哪个好免费的TK加速器推荐
TK小达人 10-01