北京网站建设公司怎么样,东莞常平网站建设,wordpress获取登录权限,wap网站 趋势分析操作系统 windows, IDE环境vscode#xff0c;本地GPU 可以新建一个jupyter文件#xff0c;运行一些测试代码
确保装好显卡驱动
在底下调出终端窗口#xff0c;默认是power shell#xff0c;我喜欢用cmd窗口 激活自己的虚拟环境#xff0c;输入命令 nvidia-smi
确保自己…操作系统 windows, IDE环境vscode本地GPU 可以新建一个jupyter文件运行一些测试代码
确保装好显卡驱动
在底下调出终端窗口默认是power shell我喜欢用cmd窗口 激活自己的虚拟环境输入命令 nvidia-smi
确保自己安装好了CUDA
!nvcc --version安装好之后输入上述代码会显示CUDA版本
安装pytorch 去官网复制他的代码运行。我用conda的代码会显示一个SSL的错误我用pip安装会给我装一个cpu版本的pytorch
import torch
print(torch.__version__)他输出结果torch2.1.0cpu就是说我装的实际上是cpu版本的。所以不要用pip,用conda 。我重装了一遍anaconda原本建的虚拟环境不会丢不用担心终于把pytorch装上去了。
检验一下
