欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

环境搭建(4)—— PyTorch在Ubutu下安装

程序员文章站 2022-03-17 14:14:39
...

环境搭建(4)—— PyTorch在Ubutu下安装


该安装主要是通过命令进行安装的。
安装前提:已经在ubutu中安装好anaconda

  1. 打开终端
  2. 查看已经安装的环境,防止重名conda env list
  3. 创建一个空的环境 conda create -n pytorch python=3.6
  4. 进入创建的环境 conda activate pytorch
  5. 安装pytorch(官网会按照配置选取,自动生成命令,复制然后执行)
    环境搭建(4)—— PyTorch在Ubutu下安装
    环境搭建(4)—— PyTorch在Ubutu下安装
  6. 因为笔者之前已经安装过tf环境,所以调试好了gpu。如果是第一次暗转够可以尝试运行命令conda install cudnn cudatoolkit numba
  7. 验证是否成功安装及GPU是否可用
# 输入python,进入python环境
python
# 在python环境下,输入以下命令
Import torch as t
Print(t.cuda.is_available())
# 如果结果显示为true,则说明支持GPU运算
  1. 完成
相关标签: pytorch