linux 18.04 cuda10.0使用pip安装pytorch

最新推荐文章于 2024-09-21 21:49:44 发布

pycharD

最新推荐文章于 2024-09-21 21:49:44 发布

阅读量1.8k

点赞数 1

分类专栏： NLP 学习文章标签： pytorch

本文链接：https://blog.csdn.net/along_000/article/details/105894537

版权

NLP 学习专栏收录该内容

6 篇文章 0 订阅

订阅专栏

linux 18.04 cuda10.0使用pip安装pytorch

背景
解决方法

背景

在pycharm中导入包import torch时报如下错误：
ModuleNotFoundError: No module named ‘torch’

解决方法

（1）下载torch-1.3.0+cu100-cp36-cp36m-linux_x86_64.whl：

study:~$ wget https://download.pytorch.org/whl/cu100/torch-1.3.0%2Bcu100-cp36-cp36m-linux_x86_64.whl

（2）下载torchvision-0.4.1+cu100-cp36-cp36m-linux_x86_64.whl：

study:~$ wget https://download.pytorch.org/whl/cu100/torchvision-0.4.1%2Bcu100-cp36-cp36m-linux_x86_64.whl

（3）在相应的环境下安装（我这里时安装在虚拟环境nlpStage中）:

(nlpStage) study:~$ ls
torch-1.3.0+cu100-cp36-cp36m-linux_x86_64.whl     
torchvision-0.4.1+cu100-cp36-cp36m-linux_x86_64.whl  


(nlpStage) study:~$ pip install torch-1.3.0+cu100-cp36-cp36m-linux_x86_64.whl 
...
Successfully installed torch-1.3.0+cu100


(nlpStage) study:~$ pip install torchvision-0.4.1+cu100-cp36-cp36m-linux_x86_64.whl
...
Successfully installed torchvision-0.4.1+cu100