安装pytorch和torchvision

1.进入https://download.pytorch.org/whl/torch_stable.html

2.nvcc -V 查看安装的cuda版本

3.根据本机cuda版本和python版本找到对应的pytorch和torchvision并下载

4.进入下载路径,pip install *.whl完成安装

你可能感兴趣的:(安装pytorch和torchvision)