安装pytorch并导入pycharm中

1.安装pytorch

在官网https://pytorch.org/上安装与自己电脑匹配的版本
安装pytorch并导入pycharm中_第1张图片
那么,如何查看本机的CUDA版本呢?
搜索NIVIDA控制面板
安装pytorch并导入pycharm中_第2张图片
安装pytorch并导入pycharm中_第3张图片
也可以:打开CMD,输入nvcc --version

2.在pycharm上导入torch

配置环境变量:按file,settings
安装pytorch并导入pycharm中_第4张图片大功告成!

你可能感兴趣的:(机器学习)