1.安装pytorch

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

2.在pycharm上导入torch

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

相关文章: