在anaconda虚拟环境里安装torch和torchvision
参考https://blog.csdn.net/youtiankeng/article/details/127759029
版本对应:在pytorch官网:https://pytorch.org/get-started/locally/
在官网那里选好cuda那些参数,用conda的命令在终端输入,能看到torch版本
比如我的:torch-1.13.1+cu117-cp37-cp37m
官网命令安装巨慢,可以先下载安装包,更快(参考https://blog.csdn.net/youtiankeng/article/details/127759029)
下载安装包:https://github.com/pytorch/vision
安装时我没用csdn blog的代码,因为照着blog,放在env根目录下显示找不到file(可能是我的问题)
报错:
WARNING: Requirement '.torch-1.13.1+cu117-cp37-cp37m-linux_x86_64.whl' looks like a filename, but the file does not exist
ERROR: Invalid requirement: '.torch==1.13.1+cu117'
最后还是在‘下载’里安装的,写一下路径就好
env-name:虚拟环境名称;xx@xx:电脑名。(可以都不用管,复制后面的代码就好)
(env-name) xx@xx:~$ pip install /home/xx/下载/torch-1.13.1+cu117-cp37-cp37m-linux_x86_64.whl
(env-name) xx@xx:~$ pip install /home/xx/下载/torchvision-0.14.1+cu117-cp37-cp37m-linux_x86_64.whl
pip install /home/xx/下载/torch-1.13.1+cu117-cp37-cp37m-linux_x86_64.whl
pip install /home/xx/下载/torchvision-0.14.1+cu117-cp37-cp37m-linux_x86_64.whl
装torchvision时候报错,一片红,估计是网络问题,多试几次,就能看到在下载,几十k的速度,等一会儿就行
后面跟着一串
File "XXXXXXXXX", line xx, in prepare_linked_requirement
类似的错误
多输入几次安装指令
前后装了好几天,终于在pycharm里import torch成功了,感人TAT
20230611 update
download from:
https://pypi.tuna.tsinghua.edu.cn/simple/torchvision/