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