pytorch安装相关

1.卸载pytorch:conda uninstall pytorch
2.卸载torchvision:conda uninstall torchvision

 

检查cuda版本:$nvcc -V

检查pytorch的cuda版本:import torch

                                         torch.version.cuda

 

安装cuda为10.0的pytorch:

             conda install pytorch torchvision cudatoolkit=10.0 -c pytorch

安装cuda为10.0的cuda:

             https://developer.nvidia.com/compute/cuda/10.0/Prod/local_installers/cuda_10.0.130_410.48_linux

             sudo sh cuda_10.0.130_410.48_linux.run

 

 

你可能感兴趣的:(pytorch安装相关)