在pycharm中安装torch_cuda
首先在网站中找到需要版本的wheel:
torch版本网站
下载到指定文件夹,我的文件放在:/home/hjw/下载/torch-2.2.1+cu121-cp38-cp38-linux_x86_64.whl
打开pycharm,在终端输入:
pip install /home/hjw/下载/torch-2.2.1+cu121-cp38-cp38-linux_x86_64.whl
首先在网站中找到需要版本的wheel:
torch版本网站
下载到指定文件夹,我的文件放在:/home/hjw/下载/torch-2.2.1+cu121-cp38-cp38-linux_x86_64.whl
打开pycharm,在终端输入:
pip install /home/hjw/下载/torch-2.2.1+cu121-cp38-cp38-linux_x86_64.whl