1.安装cuda。命令行指令(nvidia-smi)查看最高的可安装的cuda版本。
2. 根据cuda,安装对应版本pytorch。 链接: 查询网址。
3. 我目前的电脑是英伟达的显卡,所以只能用cup,不需要下载cuda,直接pytorch里面下载# CPU only对应的版本即可。
4. python终端看torch和cuda是否安装成功。
import torch
torch.cuda.is_available()
1.安装cuda。命令行指令(nvidia-smi)查看最高的可安装的cuda版本。
2. 根据cuda,安装对应版本pytorch。 链接: 查询网址。
3. 我目前的电脑是英伟达的显卡,所以只能用cup,不需要下载cuda,直接pytorch里面下载# CPU only对应的版本即可。
4. python终端看torch和cuda是否安装成功。
import torch
torch.cuda.is_available()