Shulou(Shulou.com)06/01 Report--

This article mainly introduces the solution to the failure of pytorch-gpu installation, which has a certain reference value, interested friends can refer to, I hope you can learn a lot after reading this article, let the editor take you to know about it.

About configuration

The editor used the ubuntu18 system at that time, and the gpu acceleration environment was already configured.

The following problems occurred after installing pytorch-gpu: import torchx = torch.Tensor (2P3) x_cuda = x.cuda ()

X can be normal, when the x_cuda is defined, the command line is stuck and no longer continues to output.

Reason:

At the beginning, pytorch-gpu was version 1.0, but the configured version of cuda and cudnn was lower, and the higher version of pytorch was not supported. Later, we chose to install version 0.4.1 of pytorch to solve this problem.

The current environment is torch-0.4.1-cp36 + 384 graphics card driver + cuda8.0+cudnn6.0.

Add: solve the problem of download error in conda installation pytorch-gpu version

After changing the source of Tsinghua University, the order of the official website is as follows:

Conda install pytorch=1.1.0 torchvision=0.3.0 cudatoolkit=9.0-c pytorch

However, HTTP error errors are often reported. Just delete-c pytorch.

Conda install pytorch=1.1.0 torchvision=0.3.0 cudatoolkit=9.0 thank you for reading this article carefully. I hope the article "solution to pytorch-gpu installation failure" shared by the editor will be helpful to you. At the same time, I also hope you will support us and pay attention to the industry information channel. More related knowledge is waiting for you to learn!

Welcome to subscribe "Shulou Technology Information " to get latest news, interesting things and hot topics in the IT industry, and controls the hottest and latest Internet news, technology news and IT industry trends.