在極算雲安裝pytorch0.4.1環境
阿新 • • 發佈:2018-12-26
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)