由于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要把 \ 或 \\ 改为 /