1. 程式人生 > >在極算雲安裝pytorch0.4.1環境

在極算雲安裝pytorch0.4.1環境

1.  已經安裝了anaconda。然後就可以建立你自己的環境了(你最好一定要指定個python版本,如下,

  我指定為python3.6, 如果需要的話你也可以指定其他python版本,並不是說anaconda自帶的是哪個python版本就只能用哪個python版本!)

conda create -n jjenv_pytorch python=3.6

 1.1 切換環境:

     conda activate jjenv_pytorch  (退出這個環境用 conda deactivate)

2.在和工作人員溝通後得知1070GPU裝的是cuda9,然後我去pytorch官網

 

按這個命令安裝GPU版本的pytorch 0.4.1版本。

3. 安裝torchvision

pip install -i https://pypi.tuna.tsinghua.edu.cn/simple torchvision  其中-i 是代表臨時使用的源,我這裡臨時使用清華源,因為pip install torchvision裝不了卡住了,所以臨時換了個源。

 

4.用如下幾條命令看GPU版本的pytorch是否安裝成功

In [1]: import torch
 
In [2]: torch.cuda.current_device()
Out[2]: 0
 
In [3]: torch.cuda.device(0)
Out[3]: 
<torch.cuda.device at 0x7efce0b03be0> In [4]: torch.cuda.device_count() Out[4]: 1 In [5]: torch.cuda.get_device_name(0) Out[5]: 'GeForce GTX 1070'

5. 安裝下專案需要的其他軟體(用pip install)

 [gensim]  (https://github.com/RaRe-Technologies/gensim)
[cytoolz]   (https://github.com/pytoolz/cytoolz)
[tensorboardX]  (https://github.com/lanpa/tensorboard-pytorch)
[pyrouge] (https://github.com/bheinzerling/pyrouge) (for evaluation)