如何根据CUDA版本安装对应的pytorch?
如何根据电脑配置安装pytorch?如何查看电脑的CUDA版本?
·
在官网中找到对应的安装命令
- 进入pytorch官网,找到下面截图
- 根据自己电脑的情况选择对应选项,如果电脑只有CPU,没有Nvidia显卡,Compute Platform那一项只能选CPU,如果有显卡,则需要查清楚自己电脑安装的CUDA版本。
- 将 Run this Command中的命令复制到cmd中(cmd要以管理员身份运行,否则会报错)
- 运行安装pytorch
如何查看自己电脑的CUDA版本?
- 以管理员身份打开cmd,在桌面 window 标志旁边一般有个搜索框,输入cmd,有个应用叫命令提示符,右键以管理员身份运行。在Linux中就是terminal。
- 输入nvcc -V,查看自己电脑的cuda版本,如下面截图显示的CUDA版本是11.5,Compute Platform那一项选CUDA11.3是可以的。
更多推荐
已为社区贡献1条内容
所有评论(0)