一、安装pytorch
方法一:网络好,使用pip 安装
1、到官网,找包
2、最好用pip安装,conda容易失败
pip install torch===1.5.1 torchvision===0.6.1 -f https://download.pytorch.org/whl/torch_stable.html
方法二: 本地安装
1.在网址:https://download.pytorch.org/whl/torch_stable.html,下载whl包
2.定位到安装包目录下
pip install 下载的包
3. 若torch安装成功后,spyder中无法导入,请参考下面这个博客。把文件替换一下就行了
4.此要安装cuda10.1,才能使用gpu跑程序时还需
conda install tensorflow-gpu # 自动安装cuda10.1
二、CUDA9.0 安装
参考博客:
https://blog.csdn.net/a583621671/article/details/87272803
三 服务器配置
按下面过程装完之后,请直接跑程序。torch,检验不成功不要紧。
# 安装pytorch=1.10
pip install -U https://download.pytorch.org/whl/cu90/torch-1.1.0-cp36-cp36m-linux_x86_64.whl
# 安装torchvision
pip install torchvision==0.3.0
参考博客:
https://blog.csdn.net/weixin_46773169/article/details/105687746
服务器安装torch17
# 安装
conda install pytorch torchvision torchaudio cudatoolkit=10.1 -c pytorch