深度学习环境配置
Pytorch环境
- 在网址torch-torchvision-python版本对应中查看你想要安装的torch和torchvision以及对应的python版本
- 安装对应版本的python
- 查看计算机CUDA版本,一般是在
/usr/local/
下面。 - 在网址
https://download.pytorch.org/whl/cu111
中查找所需的torch下载,其中{cu11}
可改成自己的cuda版本 - 在网址
https://download.pytorch.org/whl/cu111
中查找所需的torchvision下载,其中{cu11}
可改成自己的cuda版本