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