Win10+Ubuntu18.04双系统安装配置教程(十一)——在虚拟环境中安装pytorch框架

Win10+Ubuntu18.04双系统安装配置教程(十一)——在虚拟环境中安装pytorch框架

写 在 前 面 : 本 系 列 笔 者 亲 测 可 行 ! 文 章 配 置 如 下 \color{red}{写在前面:本系列笔者亲测可行!文章配置如下}
win10,Ubuntu18.04,nvidia驱动显示cuda版本11.7(只要高于10.2均可参考本配置教程)。
共将创建2个虚拟环境,分别取名为tensor37和torch37。
tensor37虚拟环境配置为python=3.7,tensorflow-gpu=2.2,cudatoolkit=10.1,cudnn=7.6.5;
torch37虚拟环境配置为python=3.7,pytorch=1.12.1,torchvision=0.13.1,torchaudio==0.12.1,cudatoolkit=10.2,cudnn=7.6.5。

注 意 \color{red}{注意} :虚拟环境中依赖的cuda版本可以低于nvidia驱动中的cuda版本,可以自行根据所需cuda版本在参考链接中查找对应pytorch版本

文章目录

  • Win10+Ubuntu18.04双系统安装配置教程(十一)——在虚拟环境中安装pytorch框架
    • 1. 创建名为torch37,python版本为3.7的虚拟环境
    • 2. 激活环境
    • 3. 安装框架及依赖cuda,cudnn
    • 参考链接

提示:ctrl+shift+c 复制终端文本,ctrl+shift+v 粘贴文本到终端

1. 创建名为torch37,python版本为3.7的虚拟环境

conda create -n torch37 python=3.7

2. 激活环境

conda activate torch37

3. 安装框架及依赖cuda,cudnn

conda install pytorch==1.12.1 torchvision==0.13.1 torchaudio==0.12.1 cudatoolkit=10.2 cudnn=7.6.5

参考链接

pytorch官网

你可能感兴趣的:(pytorch,ubuntu,深度学习)