由于pytorch带cuda的包下载得比较慢。所以在pytorch的网站https://download.pytorch.org/whl/torch_stable.html上用迅雷下载了要用的wheel。(注:我这里学校宿舍电脑用迅雷下载的速度比平时速度快了三倍)

但是pycharm上的终端似乎不是项目的虚拟环境,也就是说,直接在Terminal里面输入pip install xxx是不能安装到项目里的。

所以,要一个办法只安装在项目的虚拟环境里:

在项目的python console里输入:

import pip

# 绝对路径
path = "F:\\xxxxxxx\\torch-1.8.1+cu102-cp38-cp38-win_amd64.whl"

pip.main(['install', path])

或者,更为推荐的是,在Terminal里输入

# 第一个是pip3路径,一般在项目的venv/Scripts/目录下。
# 第二个路径是wheel的绝对路径。
".\venv\Scripts\pip3.exe" install "F:\xxxxxx\torch-1.8.1+cu102-cp38-cp38-win_amd64.whl"

注意,上述路径都是Windows环境下的,如果是Linux要把 \ 或 \\ 改为 /

Logo

魔乐社区(Modelers.cn) 是一个中立、公益的人工智能社区,提供人工智能工具、模型、数据的托管、展示与应用协同服务,为人工智能开发及爱好者搭建开放的学习交流平台。社区通过理事会方式运作,由全产业链共同建设、共同运营、共同享有,推动国产AI生态繁荣发展。

更多推荐