1.安装pytorch

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

2.在pycharm上导入torch

配置环境变量:按file,settings
在这里插入图片描述大功告成!

更多推荐