linux下安装pytorch的正确版本

1 查看cuda版本

一定要退出已安装的环境,到base下查看版本!

nvcc -V

2 去pytorch官网选择对应版本下载

PyTorch

2.1 找到以前的版本

2.2 直接搜对应的版本

2.3 找到对应命令

3 安装完验证

python3
import torch
print(torch.cuda.is_available())
print(torch.cuda.get_device_name(0))

4 卸载环境

conda uninstall pttorch

pip uninstall libtorch

猜你喜欢

转载自blog.csdn.net/weixin_43488159/article/details/129432060