pytorch安装

安装分两步

1.到官网,根据自己的系统找到想要的版本,如此处的v1.4.0

pytorch安装_第1张图片

2.安装对应版本的wheel(以此处的1.4.0为例)

pytorch安装_第2张图片

3.检查是否装好,不报错就好了

python #进入python环境输入以下命令
import torch
import torchvision

4.补充

sudo lshw -c video #显示显卡信息
nvcc -V #显示终端信息
python --version #查看python版本

你可能感兴趣的:(安装心得,报错解决方法,pytorch,机器学习,神经网络,深度学习)